Update app.py

app.py

@@ -46,6 +46,17 @@ 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"
+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(
+    MODEL_ID_3,
+    trust_remote_code=True,
+    torch_dtype=torch.float16 if device == "cuda" else torch.float32
+).to(device).eval()
+logger.info(f"Model '{MODEL_ID_3}' loaded successfully.")
+
 @spaces.GPU
 def parse_page(image: Image.Image, model_name: str) -> str:
     """
@@ -55,6 +66,8 @@ 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":
+        current_processor, current_model = processor_3, model_3
     else:
         raise ValueError(f"Unknown model choice: {model_name}")
 
@@ -246,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"], label="Select Model⚡️", value="Logics-Parsing")
+                model_choice = gr.Dropdown(choices=["Logics-Parsing", "Gliese-OCR-7B-Post1.0", "ERGO-7B"], 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)