init project

modules/pe3r/demo.py

@@ -549,7 +549,7 @@ def set_scenegraph_options(inputfiles, winsize, refid, scenegraph_type):
     return winsize, refid
 
 
-@spaces.GPU
+@spaces.GPU(duration=300)
 def main_demo(tmpdirname, pe3r, server_name, server_port, silent=False):
     device = 'cuda' if torch.cuda.is_available() else 'cpu'
 #    scene, outfile, imgs = get_reconstructed_scene(

modules/pe3r/models.py

@@ -14,7 +14,7 @@ import spaces
 import torch
 
 class Models:
-    @spaces.GPU
+    @spaces.GPU(duration=300)
     def __init__(self, device='cpu'):
         device = 'cuda' if torch.cuda.is_available() else 'cpu'
         # -- mast3r --