Download model

app.py

@@ -14,7 +14,12 @@ subprocess.run(
     ), check=True
 )
 
+
 if __name__ == "__main__":
+    from huggingface_hub import snapshot_download
+
+    snapshot_download("Wuvin/Unique3D", repo_type="model", local_dir="./ckpt")
+
     import os
     import sys
     sys.path.append(os.curdir)
@@ -49,7 +54,7 @@ _DESCRIPTION = '''
 
 def launch():
     model_zoo.init_models()
-        
+
     with gr.Blocks(
         title=_TITLE,
         # theme=gr.themes.Monochrome(),
@@ -61,6 +66,6 @@ def launch():
         create_3d_ui("wkl")
 
     demo.queue().launch(share=True)
-    
+
 if __name__ == '__main__':
     fire.Fire(launch)