Update model.py

model.py

@@ -7,22 +7,38 @@ import subprocess
 import sys
 import json
 
-if os.getenv("SYSTEM") == "spaces":
-    # Update pip first
-    subprocess.run(shlex.split("pip install --upgrade pip"), check=True)
-    
-    # Install specific versions of mim and click
-    subprocess.run(shlex.split("pip install mim==0.1.0"), check=True)
-    subprocess.run(shlex.split("pip install click==8.0.4"), check=True)
-
-    import mim
+def run_command(command):
+    try:
+        subprocess.run(shlex.split(command), check=True, capture_output=True, text=True)
+    except subprocess.CalledProcessError as e:
+        print(f"Error running command '{command}':")
+        print(f"Exit code: {e.returncode}")
+        print(f"stdout: {e.stdout}")
+        print(f"stderr: {e.stderr}")
+        raise
 
-    mim.uninstall("mmcv-full", confirm_yes=True)
-    mim.install("mmcv-full==1.5.0", is_yes=True)
-
-    subprocess.run(shlex.split("pip uninstall -y opencv-python"))
-    subprocess.run(shlex.split("pip uninstall -y opencv-python-headless"))
-    subprocess.run(shlex.split("pip install opencv-python-headless==4.5.5.64"))
+if os.getenv("SYSTEM") == "spaces":
+    try:
+        # Update pip first
+        run_command("pip install --upgrade pip")
+        
+        # Install openmim version 0.3.9
+        run_command("pip install openmim==0.3.9")
+        
+        # Install specific version of click
+        run_command("pip install click==8.0.4")
+
+        from mmcv.utils import install_module, uninstall_module
+
+        uninstall_module("mmcv-full")
+        install_module("mmcv-full==1.5.0")
+
+        run_command("pip uninstall -y opencv-python")
+        run_command("pip uninstall -y opencv-python-headless")
+        run_command("pip install opencv-python-headless==4.5.5.64")
+    except Exception as e:
+        print(f"An error occurred during setup: {str(e)}")
+        sys.exit(1)
 
 import huggingface_hub
 import numpy as np