Upload app_hg.py with huggingface_hub

app_hg.py

@@ -174,13 +174,13 @@ example_ts = get_example_txt_list()
 # initial models
 ################################################################
 
-worker_xbg = Removebg()
 print(f"loading {args.text2image_path}")
 worker_t2i = Text2Image(
     pretrain = args.text2image_path, 
     device = args.device, 
     save_memory = args.save_memory
 )
+worker_xbg = Removebg()
 worker_i2v = Image2Views(
     use_lite = args.use_lite, 
     device = args.device,
@@ -193,15 +193,15 @@ worker_v23 = Views2Mesh(
     device = args.device,
     save_memory = args.save_memory
 )
-worker_gif = GifRenderer(args.device)
+worker_gif = GifRenderer(device=args.device)
 
 
 if BAKE_AVAILEBLE:
-    worker_baker = MeshBaker()
+    worker_baker = MeshBaker(device=args.device)
 
 
 ### functional modules    
-@spaces.GPU(duration=180)
+@spaces.GPU
 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,6 @@ def stage_0_t2i(text, image, seed, step):
     dst = worker_xbg(image, save_folder)
     return dst, save_folder
 
-@spaces.GPU(duration=180)
 def stage_1_xbg(image, save_folder, force_remove): 
     if isinstance(image, str):
         image = Image.open(image)
@@ -235,7 +234,7 @@ def stage_1_xbg(image, save_folder, force_remove):
     rgba.save(dst)
     return dst
 
-@spaces.GPU(duration=180)
+@spaces.GPU
 def stage_2_i2v(image, seed, step, save_folder):
     if isinstance(image, str):
         image = Image.open(image)
@@ -250,7 +249,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(duration=180)
+@spaces.GPU(duration=120)
 def stage_3_v23(
     views_pil, 
     cond_pil, 
@@ -273,7 +272,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(duration=180)
+@spaces.GPU
 def stage_3p_baking(save_folder, color, bake):
     if color == "texture" and bake:
         obj_dst = worker_baker(save_folder)