Update app.py

app.py

@@ -46,8 +46,8 @@ model_2 = Qwen2_5_VLForConditionalGeneration.from_pretrained(
 ).to(device).eval()
 logger.info(f"Model '{MODEL_ID_2}' loaded successfully.")
 
-# Model 3: ERGO-7B
-MODEL_ID_3 = "nota-ai/ERGO-7B"
+# Model 3: olmOCR-7B-0825
+MODEL_ID_3 = "allenai/olmOCR-7B-0825"
 logger.info(f"Loading model 3: {MODEL_ID_3}")
 processor_3 = AutoProcessor.from_pretrained(MODEL_ID_3, trust_remote_code=True)
 model_3 = Qwen2_5_VLForConditionalGeneration.from_pretrained(
@@ -66,7 +66,7 @@ def parse_page(image: Image.Image, model_name: str) -> str:
         current_processor, current_model = processor_1, model_1
     elif model_name == "Gliese-OCR-7B-Post1.0":
         current_processor, current_model = processor_2, model_2
-    elif model_name == "ERGO-7B":
+    elif model_name == "olmOCR-7B-0825":
         current_processor, current_model = processor_3, model_3
     else:
         raise ValueError(f"Unknown model choice: {model_name}")
@@ -259,7 +259,7 @@ def main():
 
         with gr.Row(elem_classes=["main-container"]):
             with gr.Column(scale=1):
-                model_choice = gr.Dropdown(choices=["Logics-Parsing", "Gliese-OCR-7B-Post1.0", "ERGO-7B"], label="Select Model⚡️", value="Logics-Parsing")
+                model_choice = gr.Dropdown(choices=["Logics-Parsing", "Gliese-OCR-7B-Post1.0", "olmOCR-7B-0825"], label="Select Model⚡️", value="Logics-Parsing")
                 file_input = gr.File(label="Upload PDF or Image", file_types=[".pdf", ".jpg", ".jpeg", ".png"], type="filepath")
                 image_preview = gr.Image(label="Preview", type="pil", interactive=False, height=280)