Update app.py

app.py

@@ -42,6 +42,7 @@ def fig_to_pil(fig):
 model_id='microsoft/Florence-2-large'
 model = models[model_id]
 processor = processors[model_id]
+processor.to("cuda")
 
 @spaces.GPU
 def run_example(task_prompt, image, text_input=None, model_id='microsoft/Florence-2-large'):
@@ -49,7 +50,7 @@ def run_example(task_prompt, image, text_input=None, model_id='microsoft/Florenc
         prompt = task_prompt
     else:
         prompt = task_prompt + text_input
-    inputs = processor(text=prompt, images=image, return_tensors="pt").to("cuda")
+    inputs = processor(text=prompt, images=image, return_tensors="pt")
     generated_ids = model.generate(
         input_ids=inputs["input_ids"],
         pixel_values=inputs["pixel_values"],