update

app.py

@@ -1,4 +1,21 @@
 import os
+import subprocess
+import sys
+
+# Check if setup has been run
+setup_marker = ".setup_complete"
+if not os.path.exists(setup_marker):
+    print("First run detected, installing dependencies...")
+    try:
+        subprocess.check_call(["bash", "setup.sh"])
+        # Create marker file to indicate setup is complete
+        with open(setup_marker, "w") as f:
+            f.write("Setup completed")
+        print("Setup completed successfully!")
+    except subprocess.CalledProcessError as e:
+        print(f"Setup failed with error: {e}")
+        sys.exit(1)
+
 import torch
 import gradio as gr
 from typing import *
@@ -10,7 +27,7 @@ from triplaneturbo_executable.utils.mesh_exporter import export_obj
 
 # Initialize global variables
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
-ADAPTER_PATH = "/home/user/app/pretrained/triplane_turbo_sd_v1.pth"
+ADAPTER_PATH = "pretrained/triplane_turbo_sd_v1.pth" #"/home/user/app/pretrained/triplane_turbo_sd_v1.pth"
 PIPELINE = None  # Will hold our pipeline instance
 OBJ_FILE_QUEUE = deque(maxlen=100)  # Queue to store OBJ file paths
 

setup.sh

@@ -0,0 +1,11 @@
+echo "Installing PyTorch with CUDA support..."
+pip install torch==2.2.0+cu121 torchvision==0.17.0+cu121 torchaudio==2.2.0+cu121 \
+--extra-index-url https://download.pytorch.org/whl/cu121
+
+echo "Reinstalling NumPy with specific version..."
+pip install --force-reinstall -v "numpy==1.25.2"
+
+echo "Installing other requirements..."
+pip install -r requirements.txt
+
+echo "Setup completed successfully!"