Update app.py

app.py

@@ -1,4 +1,4 @@
-from transformers import MllamaForConditionalGeneration, AutoProcessor, TextIteratorStreamer , AutoModelForImageTextToText , LlavaNextProcessor ,LlavaNextForConditionalGeneration ,Qwen2VLForConditionalGeneration
+from transformers import MllamaForConditionalGeneration, AutoProcessor, TextIteratorStreamer , AutoModelForImageTextToText , LlavaNextProcessor ,LlavaNextForConditionalGeneration ,Qwen2VLForConditionalGeneration , Qwen2_5_VLForConditionalGeneration
 from PIL import Image
 import requests
 import torch
@@ -7,8 +7,8 @@ import gradio as gr
 from gradio import FileData
 import time
 import spaces
-ckpt = "mistral-community/pixtral-12b"
-model = AutoModelForImageTextToText.from_pretrained(ckpt,
+ckpt = "Qwen/Qwen2.5-VL-7B-Instruct"
+model = Qwen2_5_VLForConditionalGeneration.from_pretrained(ckpt,
     torch_dtype=torch.bfloat16, trust_remote_code=True ).to("cuda")
 processor = AutoProcessor.from_pretrained(ckpt,trust_remote_code=True)