Update app.py

app.py

@@ -186,7 +186,7 @@ def main():
         else:
             text_input = f"<grounding>{text_input}"
 
-        inputs = processor(text=text_input, images=image_input, return_tensors="pt")
+        inputs = processor(text=text_input, images=image_input, return_tensors="pt").to("cuda")
 
         generated_ids = model.generate(
             pixel_values=inputs["pixel_values"],