updates

app.py

@@ -5,11 +5,13 @@ import torch
 from PIL import Image, ImageDraw
 import gradio as gr
 
+from transformers import Qwen2VLModel
+
 from transformers import Qwen2VLForConditionalGeneration, AutoProcessor
 from qwen_vl_utils import process_vision_info  # include this file in your repo if not pip-installable
 
 # ---- model & processor loaded on CPU ----
-model = Qwen2VLForConditionalGeneration.from_pretrained(
+model = Qwen2VLModel.from_pretrained(
     "ByteDance-Seed/UI-TARS-1.5-7B",
     device_map="auto",
     torch_dtype=torch.float16,    # CPU-friendly