Update gpu

app.py

@@ -1,14 +1,17 @@
 from PIL import Image
 from unsloth import FastVisionModel
-model, tokenizer = FastVisionModel.from_pretrained(
-    model_name = "davidr99/qwen2-7b-instruct-blackjack", # YOUR MODEL YOU USED FOR TRAINING
-    load_in_4bit = True, # Set to False for 16bit LoRA
-)
-FastVisionModel.for_inference(model) # Enable for inference!
+import spaces
 
 import gradio as gr
 
+@spaces.GPU(duration=30)
 def blackjack_ai(image):
+    model, tokenizer = FastVisionModel.from_pretrained(
+        model_name = "davidr99/qwen2-7b-instruct-blackjack", # YOUR MODEL YOU USED FOR TRAINING
+        load_in_4bit = True, # Set to False for 16bit LoRA
+    )
+    FastVisionModel.for_inference(model) # Enable for inference!
+
     image = Image.fromarray(image.astype('uint8'), 'RGB')
     instruction = "Write the LaTeX representation for this image."