Upload 3 files

app.py

@@ -22,44 +22,113 @@ import sys
 import subprocess
 
 # ============================================================================
-# INSTALL MMCV/MMDET/MMENGINE VIA MIM AT RUNTIME
+# INSTALL MMCV/MMDET/MMENGINE WITH CUDA EXTENSIONS
 # ============================================================================
-# mmcv needs compiled CUDA extensions - pip install doesn't work properly.
-# We must use `mim install` to get the pre-built wheels with extensions.
+# mmcv needs pre-built CUDA extensions. We must install from OpenMMLab's
+# wheel index with the correct CUDA and PyTorch version.
 # ============================================================================
+
+def get_torch_cuda_version():
+    """Detect PyTorch and CUDA versions for wheel selection."""
+    import torch
+    
+    torch_version = torch.__version__.split('+')[0]  # e.g., "2.1.0"
+    torch_major_minor = '.'.join(torch_version.split('.')[:2])  # e.g., "2.1"
+    
+    if torch.cuda.is_available():
+        cuda_version = torch.version.cuda  # e.g., "12.1"
+        cuda_tag = 'cu' + cuda_version.replace('.', '')[:3]  # e.g., "cu121"
+    else:
+        cuda_tag = 'cpu'
+    
+    return torch_major_minor, cuda_tag
+
+
 def install_mm_packages():
-    """Install mmcv, mmdet, mmengine using mim (required for CUDA extensions)."""
-    packages = [
-        ("mmengine", "0.10.7"),
-        ("mmcv", "2.1.0"),
-        ("mmdet", "3.3.0"),
-    ]
+    """Install mmcv, mmdet, mmengine with proper CUDA extensions."""
     
-    for pkg, version in packages:
-        try:
-            __import__(pkg)
-            print(f"✅ {pkg} already installed")
-        except ImportError:
-            print(f"📦 Installing {pkg}=={version} via mim...")
-            result = subprocess.run(
-                [sys.executable, "-m", "mim", "install", f"{pkg}=={version}"],
-                capture_output=True,
-                text=True
-            )
-            if result.returncode != 0:
-                print(f"⚠️ mim install failed, trying pip: {result.stderr}")
-                # Fallback to pip for mmengine (doesn't need extensions)
-                subprocess.run(
-                    [sys.executable, "-m", "pip", "install", f"{pkg}=={version}"],
-                    capture_output=True,
-                    text=True
+    # First install mmengine (no CUDA extensions needed)
+    try:
+        import mmengine
+        print(f"✅ mmengine already installed: {mmengine.__version__}")
+    except ImportError:
+        print("📦 Installing mmengine...")
+        subprocess.run(
+            [sys.executable, "-m", "pip", "install", "mmengine==0.10.7"],
+            capture_output=True, text=True
+        )
+        print("✅ mmengine installed")
+    
+    # Install mmcv with CUDA extensions from OpenMMLab wheel index
+    try:
+        import mmcv
+        from mmcv.ops import roi_align  # Test if extensions work
+        print(f"✅ mmcv already installed with extensions: {mmcv.__version__}")
+    except (ImportError, ModuleNotFoundError) as e:
+        print(f"📦 Installing mmcv with CUDA extensions... (reason: {e})")
+        
+        # Get versions for wheel selection
+        torch_version, cuda_tag = get_torch_cuda_version()
+        print(f"   Detected: PyTorch {torch_version}, CUDA tag: {cuda_tag}")
+        
+        # OpenMMLab wheel index URL
+        wheel_index = f"https://download.openmmlab.com/mmcv/dist/{cuda_tag}/torch{torch_version}/index.html"
+        print(f"   Wheel index: {wheel_index}")
+        
+        # Uninstall any existing broken mmcv
+        subprocess.run(
+            [sys.executable, "-m", "pip", "uninstall", "mmcv", "-y"],
+            capture_output=True, text=True
+        )
+        
+        # Install from OpenMMLab wheel index
+        result = subprocess.run(
+            [sys.executable, "-m", "pip", "install", "mmcv==2.1.0", "-f", wheel_index],
+            capture_output=True, text=True
+        )
+        
+        if result.returncode != 0:
+            print(f"⚠️ First attempt failed: {result.stderr}")
+            # Try alternative CUDA versions
+            for alt_cuda in ["cu121", "cu118", "cu117"]:
+                if alt_cuda == cuda_tag:
+                    continue
+                alt_wheel_index = f"https://download.openmmlab.com/mmcv/dist/{alt_cuda}/torch{torch_version}/index.html"
+                print(f"   Trying alternative: {alt_wheel_index}")
+                result = subprocess.run(
+                    [sys.executable, "-m", "pip", "install", "mmcv==2.1.0", "-f", alt_wheel_index],
+                    capture_output=True, text=True
                 )
-            print(f"✅ {pkg} installed")
+                if result.returncode == 0:
+                    break
+        
+        print("✅ mmcv installed")
+    
+    # Install mmdet
+    try:
+        import mmdet
+        print(f"✅ mmdet already installed: {mmdet.__version__}")
+    except ImportError:
+        print("📦 Installing mmdet...")
+        subprocess.run(
+            [sys.executable, "-m", "pip", "install", "mmdet==3.3.0"],
+            capture_output=True, text=True
+        )
+        print("✅ mmdet installed")
+
 
 # Run installation before other imports
-print("🔧 Checking MM packages...")
+print("🔧 Setting up MM packages with CUDA extensions...")
 install_mm_packages()
 
+# Verify installation
+print("🔍 Verifying mmcv extensions...")
+try:
+    from mmcv.ops import roi_align
+    print("✅ mmcv._ext loaded successfully!")
+except Exception as e:
+    print(f"⚠️ Warning: mmcv extensions may not be fully loaded: {e}")
+
 # ============================================================================
 # STANDARD IMPORTS (after MM packages are installed)
 # ============================================================================
@@ -625,4 +694,8 @@ def create_demo():
 # ============================================================================
 if __name__ == "__main__":
     demo = create_demo()
-    demo.launch()
+    # Pass theme and css to launch() for Gradio 5.50+/6.0 compatibility
+    demo.launch(
+        theme=gr.themes.Soft(),
+        css=CUSTOM_CSS
+    )

requirements.txt

@@ -1,14 +1,12 @@
 # =============================================================================
 # WeDetect HuggingFace Spaces - Requirements
 # =============================================================================
-# Compatible with Gradio 5.50.0+ and huggingface_hub 1.x
-#
-# NOTE: mmcv, mmdet, mmengine are NOT listed here because they need to be
-# installed via `mim` to get proper CUDA extensions. They are installed
-# at runtime in app.py using subprocess.
+# IMPORTANT: mmcv, mmdet, mmengine are NOT listed here.
+# They are installed at runtime in app.py from OpenMMLab's wheel index
+# to ensure we get the version with compiled CUDA extensions.
 # =============================================================================
 
-# Core ML frameworks
+# Core ML frameworks (must be installed BEFORE app.py runs)
 torch>=2.0.0
 torchvision>=0.15.0
 
@@ -21,9 +19,6 @@ spaces
 Pillow>=9.0.0
 numpy>=1.21.0
 
-# OpenMIM - used to install mmcv/mmdet/mmengine at runtime
-openmim
-
 # Additional dependencies for WeDetect
 pycocotools
 supervision>=0.19.0
@@ -32,3 +27,6 @@ accelerate>=1.0.0
 terminaltables
 jsonlines
 tabulate
+
+# Note: mmcv, mmdet, mmengine are installed at runtime via:
+# pip install mmcv==2.1.0 -f https://download.openmmlab.com/mmcv/dist/{cuda}/torch{version}/index.html