Update app.py

app.py

@@ -64,8 +64,8 @@ model_o = Qwen2_5_VLForConditionalGeneration.from_pretrained(
     MODEL_ID_O, trust_remote_code=True, subfolder=SUBFOLDER, 
     torch_dtype=torch.float16).to(device).eval()
 
-# Load OpenVLThinker-7B-v1.2
-MODEL_ID_A = "ydeng9/OpenVLThinker-7B-v1.2"
+# Load ViGoRL-MCTS-SFT-7b-Spatial
+MODEL_ID_A = "gsarch/ViGoRL-MCTS-SFT-7b-Spatial"
 processor_a = AutoProcessor.from_pretrained(MODEL_ID_A, trust_remote_code=True)
 model_a = Qwen2_5_VLForConditionalGeneration.from_pretrained(
     MODEL_ID_A, trust_remote_code=True,
@@ -118,7 +118,7 @@ def generate_image(model_name: str,
     elif model_name == "MonkeyOCR-pro-1.2B":
         processor = processor_t
         model = model_t
-    elif model_name == "OpenVLThinker-7B":
+    elif model_name == "ViGoRL-MCTS-SFT-7B":
         processor = processor_a
         model = model_a
     elif model_name == "ViGoRL-MCTS-SFT-3B":
@@ -188,7 +188,7 @@ def generate_video(model_name: str,
     elif model_name == "MonkeyOCR-pro-1.2B":
         processor = processor_t
         model = model_t
-    elif model_name == "OpenVLThinker-7B":
+    elif model_name == "ViGoRL-MCTS-SFT-7B":
         processor = processor_a
         model = model_a
     elif model_name == "ViGoRL-MCTS-SFT-3B":
@@ -340,7 +340,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
 
             model_choice = gr.Radio(choices=[
                  "Camel-Doc-OCR-080125(v2)", "OCRFlux-3B",
-                 "OpenVLThinker-7B", "ViGoRL-MCTS-SFT-3B",
+                 "ViGoRL-MCTS-SFT-7B", "ViGoRL-MCTS-SFT-3B",
                  "Owlet-Safety-3B-1", "MonkeyOCR-pro-1.2B"
             ],
                                     label="Select Model",