Update app.py

app.py

@@ -35,13 +35,6 @@ model_m = Qwen2_5_VLForConditionalGeneration.from_pretrained(
     MODEL_ID_M, trust_remote_code=True,
     torch_dtype=torch.float16).to(device).eval()
 
-# Load ViGoRL-MCTS-SFT-3b-Spatial
-MODEL_ID_P = "gsarch/ViGoRL-MCTS-SFT-3b-Spatial"
-processor_p = AutoProcessor.from_pretrained(MODEL_ID_P, trust_remote_code=True)
-model_p = Qwen2_5_VLForConditionalGeneration.from_pretrained(
-    MODEL_ID_P, trust_remote_code=True,
-    torch_dtype=torch.float16).to(device).eval()
-
 # Load OCRFlux-3B
 MODEL_ID_X = "ChatDOC/OCRFlux-3B"
 processor_x = AutoProcessor.from_pretrained(MODEL_ID_X, trust_remote_code=True)
@@ -121,9 +114,6 @@ def generate_image(model_name: str,
     elif model_name == "ViGoRL-MCTS-SFT-7B":
         processor = processor_a
         model = model_a
-    elif model_name == "ViGoRL-MCTS-SFT-3B":
-        processor = processor_p
-        model = model_p
     else:
         yield "Invalid model selected.", "Invalid model selected."
         return
@@ -191,9 +181,6 @@ def generate_video(model_name: str,
     elif model_name == "ViGoRL-MCTS-SFT-7B":
         processor = processor_a
         model = model_a
-    elif model_name == "ViGoRL-MCTS-SFT-3B":
-        processor = processor_p
-        model = model_p
     else:
         yield "Invalid model selected.", "Invalid model selected."
         return
@@ -339,12 +326,11 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
                 with gr.Accordion("(Result.md)", open=False):
                     markdown_output = gr.Markdown(
                         label="markup.md")
-                #download_btn = gr.Button("Download Result.md")
 
             model_choice = gr.Radio(choices=[
                  "Camel-Doc-OCR-080125(v2)", "OCRFlux-3B",
-                 "ViGoRL-MCTS-SFT-7B", "ViGoRL-MCTS-SFT-3B",
-                 "Behemoth-3B-070225", "MonkeyOCR-pro-1.2B"
+                 "ViGoRL-MCTS-SFT-3B", "Behemoth-3B-070225",
+                 "MonkeyOCR-pro-1.2B"
             ],
                                     label="Select Model",
                                     value="Camel-Doc-OCR-080125(v2)")