Update app.py

app.py

@@ -31,8 +31,8 @@ def blackjack_ai(image):
     
     prompt = "<image><bos>extract json\n"
     image_file = Image.fromarray(image.astype('uint8'), 'RGB')
-    inputs = processor(images=[image_file], text=prompt, return_tensors="pt")
-    output = model.generate(**inputs, max_new_tokens=125)
+    inputs = processor(images=[image_file], text=prompt, return_tensors="pt").to('cuda')
+    output = model.generate(**inputs, max_new_tokens=125).to('cpu')
     return processor.decode(output[0], skip_special_tokens=True)[14:]
 
 with gr.Blocks() as demo: