Upload folder using huggingface_hub

README.md

@@ -5,7 +5,7 @@ colorFrom: purple
 colorTo: red
 sdk: gradio
 sdk_version: 4.44.1
-app_file: hg_app.py
+app_file: hg_app_bak.py
 pinned: false
 short_description: Text-to-3D and Image-to-3D Generation
 ---

hg_app.py

@@ -1,5 +1,5 @@
 # pip install gradio==4.44.1
-if True:
+if False:
     import os
     import spaces
     import subprocess

hg_app_bak.py

@@ -3,28 +3,17 @@ if False:
     import os
     import spaces
     import subprocess
-    def install_cuda_toolkit():
-        # CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run"
-        CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run"
-        CUDA_TOOLKIT_FILE = "/tmp/%s" % os.path.basename(CUDA_TOOLKIT_URL)
-        subprocess.call(["wget", "-q", CUDA_TOOLKIT_URL, "-O", CUDA_TOOLKIT_FILE])
-        subprocess.call(["chmod", "+x", CUDA_TOOLKIT_FILE])
-        subprocess.call([CUDA_TOOLKIT_FILE, "--silent", "--toolkit"])
-    
-        os.environ["CUDA_HOME"] = "/usr/local/cuda"
-        os.environ["PATH"] = "%s/bin:%s" % (os.environ["CUDA_HOME"], os.environ["PATH"])
-        os.environ["LD_LIBRARY_PATH"] = "%s/lib:%s" % (
-            os.environ["CUDA_HOME"],
-            "" if "LD_LIBRARY_PATH" not in os.environ else os.environ["LD_LIBRARY_PATH"],
-        )
-        # Fix: arch_list[-1] += '+PTX'; IndexError: list index out of range
-        os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
-    
-    install_cuda_toolkit()
+    import sys
+    import shlex
+    print("cd /home/user/app/hy3dgen/texgen/differentiable_renderer/ && bash compile_mesh_painter.sh")
     os.system("cd /home/user/app/hy3dgen/texgen/differentiable_renderer/ && bash compile_mesh_painter.sh")
-    os.system("cd /home/user/app/hy3dgen/texgen/custom_rasterizer && pip install .")
-    # os.system("cd /home/user/app/hy3dgen/texgen/custom_rasterizer && CUDA_HOME=/usr/local/cuda FORCE_CUDA=1 TORCH_CUDA_ARCH_LIST='8.0;8.6;8.9;9.0' python setup.py install")
+    print('install custom')
+    subprocess.run(shlex.split("pip install custom_rasterizer-0.1-cp310-cp310-linux_x86_64.whl"), check=True)    
+    IP = "0.0.0.0"
+    PORT = 7860
 else:
+    IP = "0.0.0.0"
+    PORT = 8080
     class spaces:
         class GPU:
             def __init__(self, duration=60):
@@ -54,7 +43,7 @@ def get_example_txt_list():
     return txt_list
 
 
-def gen_save_folder(max_size=60):
+def gen_save_folder(max_size=6000):
     os.makedirs(SAVE_DIR, exist_ok=True)
     exists = set(int(_) for _ in os.listdir(SAVE_DIR) if not _.startswith("."))
     cur_id = min(set(range(max_size)) - exists) if len(exists) < max_size else -1
@@ -410,4 +399,4 @@ if __name__ == '__main__':
     face_reduce_worker = FaceReducer()
 
     demo = build_app()
-    demo.queue().launch()
+    demo.queue().launch(server_name=IP,server_port=PORT)