Update app.py

app.py

@@ -34,7 +34,7 @@ def process_image(image, text_input=None):
     # Resize the image
     image = image.resize((new_width, new_height))
 
-    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"],