fix GPU quota

README.md

@@ -4,7 +4,7 @@ emoji: ✨
 colorFrom: yellow
 colorTo: green
 sdk: gradio
-sdk_version: 4.20.1
+sdk_version: 5.12.0
 python_version: 3.10.13
 app_file: app.py
 pinned: false

app.py

@@ -59,7 +59,7 @@ def get_dust3r_args_parser():
     return parser
 
 
-@spaces.GPU()
+@spaces.GPU(duration=150)
 def run_dust3r(inputfiles, input_path=None):
 
     if input_path is not None:
@@ -154,7 +154,7 @@ def run_dust3r(inputfiles, input_path=None):
 
     return outfile, feat_image_path, opt, None, None
 
-@spaces.GPU()
+@spaces.GPU(duration=150)
 def run_feat2gs(opt, niter=2000):
 
     if opt is None:
@@ -214,7 +214,7 @@ def run_feat2gs(opt, niter=2000):
     except Exception as e:
         raise gr.Error(f"Step 2 failed: {str(e)}")
 
-@spaces.GPU()
+@spaces.GPU(duration=150)
 def run_render(opt, args, cam_traj='ellipse'):
     if opt is None or args is None:
         raise gr.Error("Please run Steps 1 and 2 first!")

requirements.txt

@@ -1,9 +1,9 @@
 torch==2.2.0
-numpy
+numpy<2
 torchvision
 roma
 evo
-gradio>=4,<5
+gradio==5.12.0
 matplotlib
 tqdm
 opencv-python