update env

app.py

@@ -1,10 +1,10 @@
+import spaces
 import os 
 import sys 
 os.environ["PYOPENGL_PLATFORM"] = "egl"
 os.environ["MESA_GL_VERSION_OVERRIDE"] = "4.1"
 
 import gradio as gr
-#import spaces
 import cv2 
 import numpy as np 
 import torch 
@@ -28,6 +28,7 @@ from ultralytics import YOLO
 LIGHT_PURPLE=(0.25098039,  0.274117647,  0.65882353)
 STEEL_BLUE=(0.2745098, 0.5098039, 0.7058824)
 
+os.system("cd /home/user/app/hort/models/tgs/models/snowflake/pointnet2_ops_lib && python setup.py install && cd /home/user/app")
 wilor_checkpoint_path = hf_hub_download(repo_id="zerchen/hort_models", filename="wilor_final.ckpt")
 hort_checkpoint_path = hf_hub_download(repo_id="zerchen/hort_models", filename="hort_final.pth.tar")
 
@@ -196,10 +197,10 @@ with gr.Blocks(title="HORT: Monocular Hand-held Objects Reconstruction with Tran
     with gr.Row():
         example_images = gr.Examples([
             ['/home/user/app/assets/test1.png'], 
-            ['./demo_img/app/assets/test2.png'], 
-            ['./demo_img/app/assets/test3.jpg'], 
-            ['./demo_img/app/assets/test4.jpeg'],
-            ['./demo_img/app/assets/test5.jpeg'] 
+            ['/home/user/app/assets/test2.png'], 
+            ['/home/user/app/assets/test3.jpg'], 
+            ['/home/user/app/assets/test4.jpeg'],
+            ['/home/user/app/assets/test5.jpeg'] 
             ], 
             inputs=input_image)
 

requirements.txt

@@ -44,5 +44,6 @@ rich
 webdataset
 ultralytics
 dill
+git+https://github.com/facebookresearch/pytorch3d.git@stable
 git+https://github.com/zerchen/lang-segment-anything.git
 typeguard