Upload app_hg.py with huggingface_hub

app_hg.py

@@ -19,7 +19,7 @@
 # optimizer states), machine-learning model code, inference-enabling code, training-enabling code, 
 # fine-tuning enabling code and other elements of the foregoing made publicly available 
 # by Tencent in accordance with TENCENT HUNYUAN COMMUNITY LICENSE AGREEMENT.
-
+import spaces
 import os
 import warnings
 import argparse
@@ -151,6 +151,7 @@ worker_v23 = Views2Mesh(
 )
 worker_gif = GifRenderer(args.device)
 
+@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("."))
@@ -161,7 +162,7 @@ def stage_0_t2i(text, image, seed, step):
     save_folder = f'./outputs/app_output/{cur_id}'
     os.makedirs(save_folder, exist_ok=True)
 
-    dst = save_folder + '/img.png'
+    dst = os.path.join(save_folder, 'img.png')
     
     if not text:
         if image is None: 
@@ -174,7 +175,8 @@ def stage_0_t2i(text, image, seed, step):
     image.save(dst)
     dst = worker_xbg(image, save_folder)
     return dst, save_folder
-
+    
+@spaces.GPU
 def stage_1_xbg(image, save_folder): 
     if isinstance(image, str):
         image = Image.open(image)
@@ -183,6 +185,7 @@ def stage_1_xbg(image, save_folder):
     rgba.save(dst)
     return dst
     
+@spaces.GPU
 def stage_2_i2v(image, seed, step, save_folder):
     if isinstance(image, str):
         image = Image.open(image)
@@ -197,6 +200,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
 def stage_3_v23(
     views_pil, 
     cond_pil, 
@@ -219,6 +223,7 @@ def stage_3_v23(
     )
     return obj_dst, glb_dst
 
+@spaces.GPU
 def stage_4_gif(obj_dst, save_folder, do_render_gif=True):
     if not do_render_gif: return None
     gif_dst = save_folder + '/output.gif'