update app

app.py

@@ -176,8 +176,8 @@ if torch.cuda.is_available():
 print("Using device:", device)
 
 # --- Model Loading ---
-# Load Nanonets-OCR-s
-MODEL_ID_V = "nanonets/Nanonets-OCR-s"
+# Load Nanonets-OCR2-3B
+MODEL_ID_V = "nanonets/Nanonets-OCR2-3B"
 processor_v = AutoProcessor.from_pretrained(MODEL_ID_V, trust_remote_code=True)
 model_v = Qwen2_5_VLForConditionalGeneration.from_pretrained(
     MODEL_ID_V,
@@ -259,7 +259,7 @@ def generate_image(model_name: str, text: str, image: Image.Image,
     elif model_name == "Qwen2-VL-OCR-2B":
         processor = processor_x
         model = model_x
-    elif model_name == "Nanonets-OCR-s":
+    elif model_name == "Nanonets-OCR2-3B":
         processor = processor_v
         model = model_v
     elif model_name == "Aya-Vision-8B":
@@ -320,7 +320,7 @@ def generate_video(model_name: str, text: str, video_path: str,
     elif model_name == "Qwen2-VL-OCR-2B":
         processor = processor_x
         model = model_x
-    elif model_name == "Nanonets-OCR-s":
+    elif model_name == "Nanonets-OCR2-3B":
         processor = processor_v
         model = model_v
     elif model_name == "Aya-Vision-8B":
@@ -425,10 +425,10 @@ with gr.Blocks(css=css, theme=thistle_theme) as demo:
                                 ])
                     
                 model_choice = gr.Radio(
-                    choices=["olmOCR-7B-0725", "Nanonets-OCR-s", "RolmOCR-7B", 
+                    choices=["Nanonets-OCR2-3B", "olmOCR-7B-0725", "RolmOCR-7B", 
                          "Aya-Vision-8B", "Qwen2-VL-OCR-2B"],
                     label="Select Model",
-                    value="olmOCR-7B-0725"
+                    value="Nanonets-OCR2-3B"
                 )
      
     image_submit.click(