Replace Provider to Novita

app.py

@@ -4,7 +4,7 @@ import subprocess
 import sys
 import os
 
-def ensure_mmpose_installed(local_path='/data/WHAM/third-party/ViTPose'):
+def ensure_mmpose_installed(local_path='/data/wheelhouse/mmpose-0.24.0-py2.py3-none-any.whl'):
     """
     Check if 'mmpose' can be imported; if not, attempt to install it from local_path.
     local_path should contain setup.py or pyproject.toml so that pip can install it.
@@ -22,28 +22,28 @@ def ensure_mmpose_installed(local_path='/data/WHAM/third-party/ViTPose'):
     else:
         print(f"'{package_name}' not found, attempting installation...")
 
-    # If we reach here, we need to install or reinstall
-    # Check that the directory exists
-    if not os.path.isdir(local_path):
-        raise FileNotFoundError(f"Specified install directory does not exist: {local_path}")
+        # If we reach here, we need to install or reinstall
+        # Check that the directory exists
+        if not os.path.isdir(local_path):
+            raise FileNotFoundError(f"Specified install directory does not exist: {local_path}")
 
-    # Construct pip install command using the current Python executable
-    cmd = [sys.executable, "-m", "pip", "install", local_path]
-    print("Running command:", " ".join(cmd))
-    try:
-        subprocess.check_call(cmd)
-    except subprocess.CalledProcessError as e:
-        raise RuntimeError(f"Failed to install mmpose: {e}")
+        # Construct pip install command using the current Python executable
+        cmd = [sys.executable, "-m", "pip", "install", local_path]
+        print("Running command:", " ".join(cmd))
+        try:
+            subprocess.check_call(cmd)
+        except subprocess.CalledProcessError as e:
+            raise RuntimeError(f"Failed to install mmpose: {e}")
 
-    # After installation, try importing again
-    try:
-        module = importlib.import_module(package_name)
-        print(f"'{package_name}' installed and imported successfully, version: {getattr(module, '__version__', 'unknown')}")
-        return True
-    except Exception as e:
-        raise RuntimeError(f"Installed but still cannot import '{package_name}': {e}")
+        # After installation, try importing again
+        try:
+            module = importlib.import_module(package_name)
+            print(f"'{package_name}' installed and imported successfully, version: {getattr(module, '__version__', 'unknown')}")
+            return True
+        except Exception as e:
+            raise RuntimeError(f"Installed but still cannot import '{package_name}': {e}")
 
-ensure_mmpose_installed('/data/WHAM/third-party/ViTPose')
+ensure_mmpose_installed('/data/wheelhouse/mmpose-0.24.0-py2.py3-none-any.whl')
 
 
 
@@ -92,7 +92,7 @@ class PoseEstimationApp:
         if not token:
             raise RuntimeError("HF_TOKEN environment variable not set, please set the access token in deployment environment.")
         try:
-            self.client = InferenceClient(model=model_name, token=token)
+            self.client = InferenceClient(model=model_name, token=token, provider="novita")
         except Exception as e:
             raise RuntimeError(f"Failed to initialize remote inference client: {e}")