Update app.py

app.py

@@ -1,4 +1,5 @@
 import torch
+import spaces
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoProcessor
 from PIL import ImageDraw
@@ -46,6 +47,7 @@ def florence_output_to_box(output):
     return None
 
 
+@spaces.GPU
 def run_example(image, text_input, model_id="AskUI/PTA-1"):
     model = models[model_id].to(device, torch_dtype)
     processor = processors[model_id]