Update app.py

app.py

@@ -63,6 +63,7 @@ def process_image(prompt: str, img: Image.Image) -> str:
             {"type": "text", "text": prompt},
         ],
     }]
+    print(messages)
     text_prompt_for_qwen = processor.apply_chat_template(
         messages, tokenize=False, add_generation_prompt=True
     )
@@ -73,7 +74,7 @@ def process_image(prompt: str, img: Image.Image) -> str:
         videos=video_inputs,
         padding=True,
         return_tensors="pt",
-    ).to('cuda:0')
+    ).to('cuda')
     generated_ids = multi_model.generate(**inputs, max_new_tokens=1024)
     input_token_len = inputs.input_ids.shape[1]
     generated_ids_trimmed = generated_ids[:, input_token_len:]