Update app.py

app.py

@@ -1,3 +1,4 @@
+%%writefile app.py
 import os
 import base64
 import torch
@@ -13,7 +14,7 @@ from sentence_transformers import SentenceTransformer
 from PIL import Image
 from io import BytesIO
 
-from transformers import AutoProcessor, AutoModelForVision2Seq
+from transformers import AutoProcessor, Qwen3VLForConditionalGeneration
 
 # ─────────────────────────────
 # CONFIG
@@ -55,7 +56,7 @@ async def lifespan(app: FastAPI):
         trust_remote_code=True
     )
 
-    model = AutoModelForVision2Seq.from_pretrained(
+    model = Qwen3VLForConditionalGeneration.from_pretrained(
         MODEL_REPO,
         torch_dtype=torch.float32,
         device_map="cpu",