Upload app_hg.py with huggingface_hub

app_hg.py

@@ -201,7 +201,7 @@ if BAKE_AVAILEBLE:
 
 
 ### functional modules    
-@spaces.GPU
+@spaces.GPU(duration=180)
 def stage_0_t2i(text, image, seed, step):
     os.makedirs('./outputs/app_output', exist_ok=True)
     exists = set(int(_) for _ in os.listdir('./outputs/app_output') if not _.startswith("."))
@@ -226,7 +226,7 @@ def stage_0_t2i(text, image, seed, step):
     dst = worker_xbg(image, save_folder)
     return dst, save_folder
 
-@spaces.GPU
+@spaces.GPU(duration=180)
 def stage_1_xbg(image, save_folder, force_remove): 
     if isinstance(image, str):
         image = Image.open(image)
@@ -235,7 +235,7 @@ def stage_1_xbg(image, save_folder, force_remove):
     rgba.save(dst)
     return dst
 
-@spaces.GPU
+@spaces.GPU(duration=180)
 def stage_2_i2v(image, seed, step, save_folder):
     if isinstance(image, str):
         image = Image.open(image)
@@ -250,7 +250,7 @@ def stage_2_i2v(image, seed, step, save_folder):
     show_img = Image.fromarray(show_img) 
     return views_img, cond_img, show_img
 
-@spaces.GPU
+@spaces.GPU(duration=180)
 def stage_3_v23(
     views_pil, 
     cond_pil, 
@@ -273,7 +273,7 @@ def stage_3_v23(
     obj_dst = save_folder + '/mesh_vertex_colors.obj' # gradio just only can show vertex shading
     return obj_dst, glb_dst
 
-@spaces.GPU
+@spaces.GPU(duration=180)
 def stage_3p_baking(save_folder, color, bake):
     if color == "texture" and bake:
         obj_dst = worker_baker(save_folder)