fix bugs of installation

app.py

@@ -10,11 +10,11 @@ import subprocess
 
 if os.getenv('SYSTEM') == 'spaces':
     subprocess.run('pip install pyembree'.split())
-    subprocess.run('pip install rembg'.split())
+    subprocess.run('pip install git+https://github.com/YuliangXiu/rembg.git@hf'.split())
     subprocess.run(
         'pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html'.split())
     subprocess.run(
-        'pip install git+https://github.com/YuliangXiu/kaolin.git'.split())
+        'pip wheel git+https://github.com/YuliangXiu/kaolin.git -w /home/user/app && pip install /home/user/app/kaolin-0.11.0-cp38-cp38-linux_x86_64.whl', shell=True)
     subprocess.run('pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py38_cu113_pyt1110/download.html'.split())
     subprocess.run(
         'pip install git+https://github.com/Project-Splinter/human_det.git'.split())
@@ -117,7 +117,7 @@ with gr.Blocks() as demo:
 
             gr.Examples(examples=examples,
                         inputs=[inp, radio_choice],
-                        cache_examples=True,
+                        cache_examples=False,
                         fn=generate_model,
                         outputs=out_lst)
 
@@ -139,9 +139,11 @@ with gr.Blocks() as demo:
                 clear_color=[0.0, 0.0, 0.0, 0.0],  label="Refined Recon")
             out_final_download = gr.File(
                 label="Download refined clothed human mesh")
+            out_kaolin_download = gr.File(
+                label="Kaolin")
 
     out_lst = [out_smpl, out_smpl_download, out_smpl_npy_download, out_recon, out_recon_download,
-               out_final, out_final_download, out_vid, out_vid_download, overlap_inp]
+               out_final, out_final_download, out_vid, out_vid_download, overlap_inp, out_kaolin_download]
 
     btn_submit.click(fn=generate_model, inputs=[
                      inp, radio_choice], outputs=out_lst)

apps/infer.py

@@ -463,8 +463,10 @@ def generate_model(in_path, model_type):
     gc.collect()
     torch.cuda.empty_cache()
     
+    kaolin_path = "/home/user/app/kaolin-0.11.0-cp38-cp38-linux_x86_64.whl"
+    
     return [smpl_glb_path, smpl_obj_path, 
             smpl_npy_path, 
             recon_glb_path, recon_obj_path, 
             refine_glb_path, refine_obj_path, 
-            video_path, video_path, overlap_path]
+            video_path, video_path, overlap_path, kaolin_path]

requirements.txt

@@ -1,3 +1,5 @@
+pip>=21.2.4
+numpy==1.22.4
 matplotlib==3.5.0
 Pillow==9.0.0
 PyOpenGL
@@ -15,7 +17,6 @@ rtree==0.9.7
 pytorch_lightning==1.2.5
 PyMCubes
 chumpy
-rembg>=2.0.3
 opencv-python
 opencv_contrib_python
 scikit-learn