Update app.py

app.py

@@ -4,6 +4,26 @@ from gradio_litmodel3d import LitModel3D
 
 import os
 import shutil
+import subprocess
+import sys
+
+# Install local wheels at runtime
+def install_local_wheels():
+    """Install the local wheel files that couldn't be installed during Docker build."""
+    wheels_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'wheels')
+    if os.path.exists(wheels_dir):
+        wheel_files = [f for f in os.listdir(wheels_dir) if f.endswith('.whl')]
+        for wheel_file in wheel_files:
+            wheel_path = os.path.join(wheels_dir, wheel_file)
+            try:
+                subprocess.check_call([sys.executable, '-m', 'pip', 'install', wheel_path])
+                print(f"Successfully installed {wheel_file}")
+            except subprocess.CalledProcessError as e:
+                print(f"Failed to install {wheel_file}: {e}")
+
+# Install wheels before importing trellis
+install_local_wheels()
+
 os.environ['SPCONV_ALGO'] = 'native'
 from typing import *
 import torch