Update app.py

app.py

@@ -4,14 +4,20 @@ import numpy as np
 import cv2
 from PIL import Image
 from threading import Thread
-from transformers import AutoProcessor, Gemma3ForConditionalGeneration, TextIteratorStreamer
+from transformers import (
+    AutoModelForCausalLM,
+    AutoTokenizer,
+    TextIteratorStreamer,
+    Qwen2VLForConditionalGeneration,
+    AutoProcessor,
+)
 import spaces
 import time
 
 # Load Model & Processor
-MODEL_ID = "google/gemma-3-12b-it"
+MODEL_ID = "Qwen/Qwen2-VL-2B-Instruct"
 processor = AutoProcessor.from_pretrained(MODEL_ID, trust_remote_code=True)
-model = Gemma3ForConditionalGeneration.from_pretrained(
+model = Qwen2VLForConditionalGeneration.from_pretrained(
     MODEL_ID,
     trust_remote_code=True,
     torch_dtype=torch.bfloat16