Update app.py

app.py

@@ -24,17 +24,6 @@ from transformers import (
     AutoTokenizer,
 )
 
-# --- Activate Forced Dark Mode --- 
-#js_func = """
-#function refresh() {
-#    const url = new URL(window.location);
-#    if (url.searchParams.get('__theme') !== 'dark') {
-#        url.searchParams.set('__theme', 'dark');
-#        window.location.href = url.href;
-#    }
-#}
-#"""
-
 # --- Constants and Model Setup ---
 MAX_INPUT_TOKEN_LENGTH = 4096
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -89,18 +78,6 @@ model_i = Qwen2_5_VLForConditionalGeneration.from_pretrained(
     MODEL_ID_I, trust_remote_code=True, torch_dtype=torch.float16
 ).to(device).eval()
 
-# Load typhoon-ocr-3b
-MODEL_ID_J = "scb10x/typhoon-ocr-3b"
-processor_j = AutoProcessor.from_pretrained(
-    MODEL_ID_J,
-    trust_remote_code=True
-)
-model_j = Qwen2_5_VLForConditionalGeneration.from_pretrained(
-    MODEL_ID_J,
-    trust_remote_code=True,
-    torch_dtype=torch.float16
-).to(device).eval()
-
 # --- Utility Functions ---
 def layoutjson2md(layout_data: List[Dict]) -> str:
     """Converts the structured JSON from Layout Analysis into formatted Markdown."""
@@ -151,7 +128,6 @@ def process_document_stream(model_name: str, task_choice: str, image: Image.Imag
     elif model_name == "Nanonets-OCR-s": processor, model = processor_c, model_c
     elif model_name == "MonkeyOCR-Recognition": processor, model = processor_g, model_g
     elif model_name == "olmOCR-7B-0725": processor, model = processor_i, model_i
-    elif model_name == "typhoon-ocr-3b": processor, model = processor_j, model_j
     else:
         yield "Invalid model selected.", "Invalid model selected.", None
         return
@@ -202,7 +178,6 @@ def create_gradio_interface():
     .process-button { border: none !important; color: white !important; font-weight: bold !important; background-color: blue !important;} 
     .process-button:hover { background-color: darkblue !important; transform: translateY(-2px) !important; box-shadow: 0 4px 8px rgba(0,0,0,0.2) !important; }
     """
-    #with gr.Blocks(theme="bethecloud/storj_theme", css=css, js=js_func) as demo:
     with gr.Blocks(theme="bethecloud/storj_theme", css=css) as demo:
         gr.HTML("""
         <div class="title" style="text-align: center">
@@ -221,8 +196,7 @@ def create_gradio_interface():
                              "MonkeyOCR-Recognition", 
                              "olmOCR-7B-0725",
                              "Nanonets-OCR-s", 
-                             "Megalodon-OCR-Sync-0713",
-                             "typhoon-ocr-3b"
+                             "Megalodon-OCR-Sync-0713"
                             ],
                     label="Select Model", value="Nanonets-OCR-s"
                 )
@@ -280,4 +254,4 @@ def create_gradio_interface():
 
 if __name__ == "__main__":
     demo = create_gradio_interface()
-    demo.queue().launch(ssr_mode=False, show_error=True)
+    demo.queue(max_size=40).launch(share=True, mcp_server=True, ssr_mode=False, show_error=True)