pre install

Dockerfile

@@ -25,4 +25,5 @@ WORKDIR $HOME/app
 # Copy the current directory contents into the container at $HOME/app setting the owner to the user
 COPY --chown=user . $HOME/app
 
+CMD ["python3", "install.py"]
 CMD ["python3", "launch.py"]

install.py

@@ -0,0 +1,21 @@
+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

@@ -16,8 +16,8 @@ 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")
+# 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