Update app.py

app.py

@@ -1,14 +1,15 @@
 import gradio as gr
-from transformers import AutoProcessor, Qwen2_5_VLForConditionalGeneration, TextIteratorStreamer
+from transformers import Qwen2VLForConditionalGeneration, AutoTokenizer, AutoProcessor, TextIteratorStreamer
+# from qwen_vl_utils import process_vision_info
 from transformers.image_utils import load_image
 from threading import Thread
 import time
 import torch
 import spaces
 
-MODEL_ID = "Qwen/Qwen2.5-VL-7B-Instruct"
+MODEL_ID = "Qwen/Qwen2-VL-2B-Instruct"
 processor = AutoProcessor.from_pretrained(MODEL_ID, trust_remote_code=True)
-model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
+model = Qwen2VLForConditionalGeneration.from_pretrained(
     MODEL_ID,
     trust_remote_code=True,
     torch_dtype=torch.bfloat16
@@ -86,7 +87,7 @@ examples = [
 
 demo = gr.ChatInterface(
     fn=model_inference,
-    description="# **Qwen2.5-VL-7B-Instruct**",
+    description="# **Qwen2-VL-2B-Instruct**",
     examples=examples,
     textbox=gr.MultimodalTextbox(label="Query Input", file_types=["image"], file_count="multiple"),
     stop_btn="Stop Generation",