Update app.py

app.py

@@ -16,6 +16,7 @@ import cv2
 from transformers import (
     Qwen2VLForConditionalGeneration,
     Qwen2_5_VLForConditionalGeneration,
+    Gemma3nForConditionalGeneration,
     AutoModelForImageTextToText,
     AutoProcessor,
     TextIteratorStreamer,
@@ -57,9 +58,9 @@ model_t = Qwen2_5_VLForConditionalGeneration.from_pretrained(
 ).to(device).eval()
 
 # Load olmOCR-7B-0225-preview
-MODEL_ID_O = "allenai/olmOCR-7B-0225-preview"
+MODEL_ID_O = "google/gemma-3n-E4B-it"
 processor_o = AutoProcessor.from_pretrained(MODEL_ID_O, trust_remote_code=True)
-model_o = Qwen2VLForConditionalGeneration.from_pretrained(
+model_o = Gemma3nForConditionalGeneration.from_pretrained(
     MODEL_ID_O,
     trust_remote_code=True,
     torch_dtype=torch.float16