update

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-# import spaces
+import spaces
 import os
 import subprocess
 
@@ -22,7 +22,7 @@ def install_cuda_toolkit():
     # CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run"
     CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run"
     CUDA_TOOLKIT_FILE = "/tmp/%s" % os.path.basename(CUDA_TOOLKIT_URL)
-    subprocess.call(["wget", CUDA_TOOLKIT_URL, "-O", CUDA_TOOLKIT_FILE])
+    subprocess.call(["wget", "-q", CUDA_TOOLKIT_URL, "-O", CUDA_TOOLKIT_FILE])
     subprocess.call(["chmod", "+x", CUDA_TOOLKIT_FILE])
     subprocess.call([CUDA_TOOLKIT_FILE, "--silent", "--toolkit"])
 
@@ -49,7 +49,7 @@ from hawor.utils.process import get_mano_faces, run_mano, run_mano_left
 from lib.eval_utils.custom_utils import load_slam_cam
 from lib.vis.run_vis2 import run_vis2_on_video, run_vis2_on_video_cam
 
-
+@spaces.GPU()
 def render_reconstruction(input_video, img_focal): 
     args = EasyDict()
     args.video_path = input_video
@@ -143,12 +143,7 @@ def render_reconstruction(input_video, img_focal):
         # run_vis2_on_video_cam(left_dict, right_dict, output_pth, img_focal, image_names, R_w2c=R_w2c_sla_all[vis_start:vis_end], t_w2c=t_w2c_sla_all[vis_start:vis_end])
         raise NotImplementedError
 
-    return vis_video_path  
-
-# @spaces.GPU()
-def run_wilow_model(image, conf, IoU_threshold=0.5):
-    img_cv2 = image[...,::-1]
-    return img_vis.astype(np.float32)/255.0, len(detections), None       
+    return vis_video_path