Update app.py

app.py

@@ -96,45 +96,6 @@ except Exception as e:
     processor_x = None
     print(f"✗ Nanonets-OCR2-3B: Failed to load - {str(e)}")
 
-
-
-
-# Load Dots.OCR - UPDATED with snapshot_download and device_map="auto"
-try:
-    MODEL_ID_D = "rednote-hilab/dots.ocr"
-    model_path_d = os.path.join(CACHE_DIR, "dots-ocr-local")
-    
-    # Download and cache model locally
-    snapshot_download(
-        repo_id=MODEL_ID_D,
-        local_dir=model_path_d,
-        local_dir_use_symlinks=False,  # Avoid symlink issues on HF Spaces
-        allow_patterns=["*.json", "*.bin", "*.safetensors", "*.txt"]
-    )
-    
-    processor_d = AutoProcessor.from_pretrained(
-        model_path_d,
-        trust_remote_code=True
-    )
-    
-    model_d = AutoModelForCausalLM.from_pretrained(
-        model_path_d,
-        attn_implementation="flash_attention_2",
-        torch_dtype=torch.bfloat16,
-        device_map="auto",  # Better memory management
-        trust_remote_code=True
-    ).eval()
-    print("✓ Dots.OCR loaded")
-except Exception as e:
-    model_d = None
-    processor_d = None
-    print(f"✗ Dots.OCR: Failed to load - {str(e)}")
-    import traceback
-    traceback.print_exc()
-
-
-
-
 # Load olmOCR-2-7B-1025
 try:
     MODEL_ID_M = "allenai/olmOCR-2-7B-1025"
@@ -198,12 +159,6 @@ def generate_image(model_name: str, text: str, image: Image.Image,
             return
         processor = processor_v
         model = model_v
-    elif model_name == "Dots.OCR":
-        if model_d is None:
-            yield "Dots.OCR is not available.", "Dots.OCR is not available."
-            return
-        processor = processor_d
-        model = model_d
     else:
         yield "Invalid model selected.", "Invalid model selected."
         return
@@ -216,8 +171,6 @@ def generate_image(model_name: str, text: str, image: Image.Image,
         return
 
 
-
-
     try:
         # Prepare messages in chat format
         messages = [{
@@ -319,11 +272,6 @@ if __name__ == "__main__":
     if model_v is not None:
         available_models.append("Chandra-OCR")
         print("  Added: Chandra-OCR")
-    if model_d is not None:
-        available_models.append("Dots.OCR")
-        print("  Added: Dots.OCR")
-
-
     if not available_models:
         print("ERROR: No models were loaded successfully!")
         exit(1)
@@ -403,7 +351,6 @@ if __name__ == "__main__":
         - **olmOCR-2-7B-1025**: Allen AI's OCR model
         - **Nanonets-OCR2-3B**: Nanonets OCR model
         - **Chandra-OCR**: Datalab OCR model
-        - **Dots.OCR**: Stranger Vision OCR model (Updated)
         """)