Update app.py

app.py

@@ -136,7 +136,7 @@ if torch.cuda.is_available():
 
 print("Using device:", device)
 
-MODEL_ID_X = "nota-ai/ERGO-7B"
+MODEL_ID_X = "Senqiao/VisionThink-Efficient"
 processor_x = AutoProcessor.from_pretrained(MODEL_ID_X, trust_remote_code=True, use_fast=False)
 model_x = Qwen2_5_VLForConditionalGeneration.from_pretrained(
     MODEL_ID_X,
@@ -179,7 +179,7 @@ model_v4 = AutoModel.from_pretrained(
 tokenizer_v4 = AutoTokenizer.from_pretrained(MODEL_ID_V4, trust_remote_code=True, use_fast=False)
 
 MODELS = {
-    "ERGO-7B": (processor_x, model_x),
+    "VisionThink-Efficient": (processor_x, model_x),
     "Typhoon-OCR-3B": (processor_t, model_t),
     "olmOCR-7B-0225-preview": (processor_o, model_o),
     "Lumian-VLR-7B-Thinking": (processor_j, model_j),
@@ -357,7 +357,7 @@ with gr.Blocks(theme=light_salmon_theme, css=css) as demo:
                             ])
 
             model_choice = gr.Radio(
-                choices=["Lumian-VLR-7B-Thinking", "ERGO-7B", "openbmb/MiniCPM-V-4", "Typhoon-OCR-3B", "olmOCR-7B-0225-preview"],
+                choices=["Lumian-VLR-7B-Thinking", "VisionThink-Efficient", "openbmb/MiniCPM-V-4", "Typhoon-OCR-3B", "olmOCR-7B-0225-preview"],
                 label="Select Model",
                 value="Lumian-VLR-7B-Thinking"
             )