Update app.py

app.py

@@ -1,4 +1,4 @@
-import gradio as gr
+iimport gradio as gr
 import torch
 from transformers import AutoProcessor, AutoModelForCausalLM, AutoTokenizer
 import time
@@ -99,11 +99,11 @@ def test_image_text_generation():
         
         results.append(f"Loading processor and model from {model_id}...")
         
-        # Load processor and model
-        from transformers import AutoProcessor, AutoModelForCausalLM
+        # Load processor and model with the correct classes
+        from transformers import Qwen2VLProcessor, Qwen2VLForConditionalGeneration
         
-        processor = AutoProcessor.from_pretrained(model_id)
-        model = AutoModelForCausalLM.from_pretrained(
+        processor = Qwen2VLProcessor.from_pretrained(model_id)
+        model = Qwen2VLForConditionalGeneration.from_pretrained(
             model_id,
             torch_dtype=torch.bfloat16,
             device_map="auto"