Disable SSR via env var GRADIO_SSR_MODE=false

app.py

@@ -2,6 +2,9 @@ import os
 import sys
 import subprocess
 
+# Disable SSR mode - it causes issues with Zero GPU
+os.environ["GRADIO_SSR_MODE"] = "false"
+
 # Check if we're running on HF Spaces with Zero GPU
 # HF Spaces sets SPACE_ID environment variable
 IS_HF_SPACE = bool(os.environ.get("SPACE_ID"))
@@ -25,12 +28,21 @@ import gradio as gr
 
 
 def _launch():
-    # Add CropFormer to PYTHONPATH (needed when using prebuilt detectron2)
+    # Add CropFormer and DeepLab to PYTHONPATH (needed when using prebuilt detectron2)
+    # Prebuilt detectron2 doesn't include projects/, so we need to add them manually
     repo_root = os.path.dirname(os.path.abspath(__file__))
-    cropformer_path = os.path.join(repo_root, "MaskClustering/third_party/detectron2/projects/CropFormer")
+    projects_root = os.path.join(repo_root, "MaskClustering/third_party/detectron2/projects")
+
+    # Add CropFormer (main project we use)
+    cropformer_path = os.path.join(projects_root, "CropFormer")
     if cropformer_path not in sys.path:
         sys.path.insert(0, cropformer_path)
 
+    # Add DeepLab (required by CropFormer for add_deeplab_config)
+    deeplab_path = os.path.join(projects_root, "DeepLab")
+    if deeplab_path not in sys.path:
+        sys.path.insert(0, deeplab_path)
+
     # Set environment variables for Zero GPU optimization
     if IS_ZERO_GPU:
         print("🚀 Detected Zero GPU environment, optimizing settings...")