Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import torch
 import requests
+import spaces
 
 from PIL import Image
 
@@ -19,13 +20,12 @@ finetuned_model_id = "davidr99/finetuned_paligemma2_blackjack"
 
 import gradio as gr
 
-model = PaliGemmaForConditionalGeneration.from_pretrained(model_id)
-model.load_adapter(finetuned_model_id)
-
-processor = AutoProcessor.from_pretrained(model_id)
-
+@spaces.GPU(duration=30)
 def blackjack_ai(image):
-    global model, processor
+    model = PaliGemmaForConditionalGeneration.from_pretrained(model_id)
+    model.load_adapter(finetuned_model_id)
+    
+    processor = AutoProcessor.from_pretrained(model_id)
     
     prompt = "<image><bos>extract json\n"
     image_file = Image.fromarray(image.astype('uint8'), 'RGB')