Spaces:

davidr99
/

qwen2-vl-7b-blackjack

Runtime error

davidr99 commited on Jan 1

Commit

9c8aa54

verified ·

1 Parent(s): 0567922

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ from qwen_vl_utils import process_vision_info
 @spaces.GPU(duration=30)
 def blackjack_ai(image):
-    model = Qwen2VLForConditionalGeneration.from_pretrained(MODEL_ID, torch_dtype="auto")
     processor = AutoProcessor.from_pretrained(MODEL_ID)
     instruction = "extract json from this image."

 @spaces.GPU(duration=30)
 def blackjack_ai(image):
+    model = Qwen2VLForConditionalGeneration.from_pretrained(MODEL_ID, torch_dtype="auto", device="auto")
     processor = AutoProcessor.from_pretrained(MODEL_ID)
     instruction = "extract json from this image."