Update app.py

app.py

@@ -2,14 +2,14 @@ import spaces
 import os
 import time
 import torch
-from transformers import AutoProcessor, AutoModelForImageTextToText
+from transformers import AutoProcessor, AutoModelForVision2Seq
 import gradio as gr
 from threading import Thread
 from PIL import Image
 
 # Model and processor initialization
 processor = AutoProcessor.from_pretrained("Qwen/QVQ-72B-Preview")
-model = AutoModelForImageTextToText.from_pretrained("Qwen/QVQ-72B-Preview").cuda().eval()
+model = AutoModelForVision2Seq.from_pretrained("Qwen/QVQ-72B-Preview").cuda().eval()
 
 # Footer
 footer = """