add zero gpu

app.py

@@ -12,6 +12,7 @@ from datetime import datetime
 from math import pi
 
 import gradio as gr
+import spaces
 import numpy as np
 import torch
 import trimesh
@@ -89,6 +90,7 @@ def instantiate_camera(camera_name, params, device):
     return eval(camera_name)(params=torch.tensor(params).float()).to(device)
 
 
+@spaces.GPU(duration=120)
 def run_model(target_dir, model_name, camera_name, params, efficiency):
 
     print("Instantiating model and camera...")
@@ -112,6 +114,7 @@ def run_model(target_dir, model_name, camera_name, params, efficiency):
     return outputs
 
 
+@spaces.GPU(duration=120)
 def gradio_demo(
     target_dir,
     model_name,
@@ -813,4 +816,4 @@ if __name__ == "__main__":
             outputs=[fx, fy, cx, cy, k1, k2, k3, k4, k5, k6, t1, t2, hfov],
         )
 
-        demo.queue(max_size=20).launch(show_error=True, share=True, ssr_mode=False)
+        demo.queue(max_size=20).launch(show_error=True, ssr_mode=False)

requirements.txt

@@ -65,6 +65,7 @@ sentry-sdk
 setproctitle
 six
 smmap
+spaces
 sympy
 tables
 tabulate

requirements_demo.txt

@@ -65,6 +65,7 @@ sentry-sdk
 setproctitle
 six
 smmap
+spaces
 sympy
 tables
 tabulate