Update app.py

app.py

@@ -12,11 +12,9 @@ MODEL_DIR = os.path.join(HOME, "models")
 print("--- Starting MinerU Setup ---")
 
 # 2. Download Models (if not present)
-# Note: This might take a few minutes on the first start!
 if not os.path.exists(MODEL_DIR):
     print(f"Downloading models to {MODEL_DIR}...")
     try:
-        # Download core models
         snapshot_download(
             "opendatalab/PDF-Extract-Kit-1.0",
             local_dir=MODEL_DIR,
@@ -30,13 +28,15 @@ else:
     print("Models found. Skipping download.")
 
 # 3. Generate magic-pdf.json Config
-# MinerU requires this file to know where the models are.
+# FIX: Define REAL_MODEL_DIR before using it
+REAL_MODEL_DIR = os.path.join(MODEL_DIR, "models")
+
 config_data = {
-    "models-dir": REAL_MODEL_DIR,
-    "device-mode": "cpu",  # Change to "cuda" if you are using a GPU Space
+    "models-dir": REAL_MODEL_DIR,  # Now this variable exists
+    "device-mode": "cpu",
     "table-config": {
         "model": "TableMaster",
-        "is_table_recog_enable": False, # Disable table recognition for speed on CPU
+        "is_table_recog_enable": False,
         "max_time": 400
     }
 }
@@ -48,11 +48,10 @@ with open(CONFIG_FILE, "w") as f:
 # 4. Launch the MinerU REST API
 print("Launching MinerU REST API...")
 
-# Change the command to use mineru-api instead of mineru-gradio
 command = [
     "mineru-api",
     "--host", "0.0.0.0",
-    "--port", "7860"  # HF Spaces requires port 7860
+    "--port", "7860"
 ]
 
 subprocess.run(command)