demo fixed

gradio_app.py → app.py

@@ -13,6 +13,7 @@ import numpy as np
 import plotly.graph_objs as go
 from my.utils.seed import seed_everything
 import os
+import wget
 
 SHARED_UI_WARNING = f'''### [NOTE]  Training may be very slow in this shared UI.
 You can duplicate and use it with a paid private GPU.
@@ -228,6 +229,20 @@ if __name__ == '__main__':
     parser = argparse.ArgumentParser()
     parser.add_argument('--share', action='store_true', help="public url")
     args = parser.parse_args()
+
+    weights_dir = './weights'
+    if not os.path.exists(weights_dir):
+        os.makedirs(weights_dir)
+    weights_path = os.path.join(weights_dir, '3DFuse_sparse_depth_injector.ckpt')
+
+    # 파일이 존재하지 않으면 wget으로 다운로드하여 저장
+    if not os.path.isfile(weights_path):
+        url = 'https://huggingface.co/jyseo/3DFuse_weights/resolve/main/models/3DFuse_sparse_depth_injector.ckpt'
+        wget.download(url, weights_path)
+        print(f'{weights_path} downloaded.')
+    else:
+        print(f'{weights_path} already exists.')
+
     
     model = None
     intermediate = Intermediate()

requirements.txt

@@ -26,6 +26,7 @@ click
 easydict
 tabulate
 plotly
+wget
 git+https://github.com/openai/point-e
 git+https://github.com/huggingface/diffusers
 git+https://github.com/cloneofsimo/lora.git

run_3DFuse.py

@@ -116,7 +116,7 @@ def fuse_3d(
     depth_weight, var_red, exp_dir, points, is_gradio, **kwargs
 ):
     del kwargs
-    
+
     if is_gradio:
         yield gr.update(visible=True), "LoRA layers tuning has just finished. \nScore distillation has started.", gr.update(visible=True)
     assert model.samps_centered()
@@ -230,7 +230,7 @@ def fuse_3d(
         out=evaluate(model, vox, poser)
         
         if is_gradio:    
-            yield gr.update(visible=True), f"Generation complete. Please check the video below. \nThe result files and logs are located at {exp_dir}", gr.update(value=out)
+            yield gr.update(visible=False), f"Generation complete. Please check the video below.", gr.update(value=out)
         else :
             yield None