Update app.py

app.py

@@ -49,7 +49,7 @@ def model_inference(
             ]
     prompt = processor.apply_chat_template(resulting_messages, add_generation_prompt=True)
     inputs = processor(text=prompt, images=[images], return_tensors="pt")
-    inputs = {k: v.to("cuda") for k, v in inputs.items()}
+    inputs = inputs.to('cuda')
     generation_args = {
         "input_ids": inputs.input_ids,
         "pixel_values": inputs.pixel_values,