Update app.py

app.py

@@ -109,7 +109,7 @@ def prep_for_rayst3r(img,depth_dict,mask):
     # save image
     save_tensor_as_png(torch.from_numpy(img), os.path.join(input_dir, "rgb.png"))
 
-@GPU(duration = 120)
+@GPU(duration = 110)
 def rayst3r_to_glb(img,depth_dict,mask,max_total_points=10e6,rotated=False):
     prep_for_rayst3r(img,depth_dict,mask)
     
@@ -160,14 +160,14 @@ def input_to_glb(outdir,img,depth_dict,mask,rotated=False):
     scene.export(outfile)
     return outfile
 
-@GPU(duration = 10)
+# @GPU(duration = 50)
 def depth_moge(input_img):
     
     input_img_torch = torch.tensor(input_img / 255, dtype=torch.float32, device=device).permute(2, 0, 1)
     output = moge_model.infer(input_img_torch)
     return output 
 
-@GPU(duration = 10)
+# @GPU(duration = 20)
 def mask_rembg(input_img):
     #masked_img = rembg.remove(input_img,)
     output_img = rembg.remove(input_img, alpha_matting=False, post_process_mask=True)
@@ -186,7 +186,7 @@ def mask_rembg(input_img):
     rgb = output_np[:,:,:3]
     return mask, rgb 
 
-@GPU(duration = 10)
+@GPU(duration = 60)
 def process_image(input_img):
 
     rotated = False