env fix

Dockerfile

@@ -8,6 +8,7 @@ RUN pip3 install --no-cache-dir \
     numpy \
     trimesh \
     Pillow \
+    bpy \
     gradio==4.40.0
 
 # Set up a new user named "user" with user ID 1000

install.py

@@ -1,21 +0,0 @@
-import os
-import sys
-import json
-import uuid
-import numpy as np
-import gradio as gr
-import trimesh
-import zipfile
-import subprocess
-from datetime import datetime
-from functools import partial
-from PIL import Image, ImageChops
-from huggingface_hub import snapshot_download
-
-
-is_local_run = os.path.exists("../SpaRP_API")
-code_dir = snapshot_download("sudo-ai/SpaRP_API", token=os.environ['HF_TOKEN']) if not is_local_run else "../SpaRP_API"
-
-os.system(f"pip install {code_dir}/gradio_model3dcolor-0.0.1-py3-none-any.whl")
-os.system(f"pip install {code_dir}/gradio_model3dnormal-0.0.1-py3-none-any.whl")
-

launch.py

@@ -1,6 +1,4 @@
 import os
-# os.system(f"pip install gradio_model3dcolor-0.0.1-py3-none-any.whl")
-# os.system(f"pip install gradio_model3dnormal-0.0.1-py3-none-any.whl")
 import sys
 import json
 import uuid
@@ -14,16 +12,11 @@ from functools import partial
 from PIL import Image, ImageChops
 from huggingface_hub import snapshot_download
 
-
-is_local_run = os.path.exists("../SpaRP_API")
-code_dir = snapshot_download("sudo-ai/SpaRP_API", token=os.environ['HF_TOKEN']) if not is_local_run else "../SpaRP_API"
-
-# os.system(f"pip install {code_dir}/gradio_model3dcolor-0.0.1-py3-none-any.whl")
-# os.system(f"pip install {code_dir}/gradio_model3dnormal-0.0.1-py3-none-any.whl")
-
 from gradio_model3dcolor import Model3DColor
 from gradio_model3dnormal import Model3DNormal
 
+is_local_run = os.path.exists("../SpaRP_API")
+code_dir = snapshot_download("sudo-ai/SpaRP_API", token=os.environ['HF_TOKEN']) if not is_local_run else "../SpaRP_API"
 
 if not is_local_run:
     zip_file_path = f'{code_dir}/examples.zip'