install specific torch

app.py

@@ -139,7 +139,7 @@ def inference(runner: DemoRunner, cfg: ProjectConfig, rgb, mask_hum, mask_obj, s
 
 
 @hydra.main(config_path='configs', config_name='configs', version_base='1.1')
-def main(cfg: ProjectConfig, args):
+def main(cfg: ProjectConfig):
     # Setup model
     runner = DemoRunner(cfg)
 
@@ -152,7 +152,8 @@ def main(cfg: ProjectConfig, args):
     with demo:
         gr.Markdown(md_description)
         gr.HTML("""<h1 style="text-align:center; color:#10768c">HDM Demo</h1>""")
-        gr.HTML("""<h3 style="text-align:center; color:#10768c">Instruction: Upload RGB, human, object masks and then click reconstruct.</h1>""")
+        gr.HTML("""<p style="text-align:center; color:#10768c">Instruction: Upload RGB, human, object masks and then click reconstruct.</p>""")
+        gr.HTML("""<p style="text-align:center; color:#10768c">You can use these methods to obtain the masks: </p>""")
 
         # Input data
         with gr.Row():
@@ -200,12 +201,12 @@ def main(cfg: ProjectConfig, args):
 
     # demo.launch(share=True)
     # Enabling queue for runtime>60s, see: https://github.com/tloen/alpaca-lora/issues/60#issuecomment-1510006062
-    demo.queue().launch(share=args.share)
+    demo.queue().launch()
 
 if __name__ == '__main__':
     from argparse import ArgumentParser
-    parser = ArgumentParser()
-    parser.add_argument('-share', default=False, action='store_true', help='allow a temporal public url')
-    args = parser.parse_args()
+    # parser = ArgumentParser()
+    # parser.add_argument('-share', default=False, action='store_true', help='allow a temporal public url')
+    # args = parser.parse_args()
 
-    main(args)
+    main()

pre-requirements.txt

@@ -1,3 +1,7 @@
+--extra-index-url https://download.pytorch.org/whl/cu113
+torch==1.11.0+cu113
+torchvision==0.12.0+cu113
+torchaudio==0.11.0
 accelerate
 diffusers
 hydra-core
@@ -8,11 +12,11 @@ plotly
 rich
 scipy
 timm
-torch
 torch-ema
 tqdm
 transformers
 wandb
 trimesh
 gradio
-
+fvcore
+iopath

requirements.txt

@@ -14,4 +14,4 @@ transformers
 wandb
 trimesh
 gradio
-pytorch3d@git+https://github.com/facebookresearch/pytorch3d.git
+https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py38_cu113_pyt1110/pytorch3d-0.7.2-cp38-cp38-linux_x86_64.whl