Spaces:
Running
on
L40S
Running
on
L40S
Upload app_hg.py with huggingface_hub
Browse files
app_hg.py
CHANGED
@@ -207,7 +207,7 @@ def gen_save_folder(max_size=30):
|
|
207 |
print(f"mkdir {save_folder} suceess !!!")
|
208 |
return save_folder
|
209 |
|
210 |
-
@
|
211 |
def stage_0_t2i(text, seed, step, save_folder):
|
212 |
dst = save_folder + '/img.png'
|
213 |
image = worker_t2i(text, seed, step)
|
@@ -225,7 +225,7 @@ def stage_1_xbg(image, save_folder, force_remove):
|
|
225 |
rgba.save(dst)
|
226 |
return dst
|
227 |
|
228 |
-
@
|
229 |
def stage_2_i2v(image, seed, step, save_folder):
|
230 |
if isinstance(image, str):
|
231 |
image = Image.open(image)
|
@@ -263,7 +263,7 @@ def stage_3_v23(
|
|
263 |
obj_dst = save_folder + '/mesh_vertex_colors.obj' # gradio just only can show vertex shading
|
264 |
return obj_dst, glb_dst
|
265 |
|
266 |
-
@
|
267 |
def stage_3p_baking(save_folder, color, bake, force, front, others, align_times):
|
268 |
if color == "texture" and bake:
|
269 |
obj_dst = worker_baker(save_folder, force, front, others, align_times)
|
@@ -272,7 +272,7 @@ def stage_3p_baking(save_folder, color, bake, force, front, others, align_times)
|
|
272 |
else:
|
273 |
return None
|
274 |
|
275 |
-
@
|
276 |
def stage_4_gif(save_folder, color, bake, render):
|
277 |
if not render: return None
|
278 |
baked_fld_list = sorted(glob(save_folder + '/view_*/bake/mesh.obj'))
|
|
|
207 |
print(f"mkdir {save_folder} suceess !!!")
|
208 |
return save_folder
|
209 |
|
210 |
+
@spaces.GPU(duration=120)
|
211 |
def stage_0_t2i(text, seed, step, save_folder):
|
212 |
dst = save_folder + '/img.png'
|
213 |
image = worker_t2i(text, seed, step)
|
|
|
225 |
rgba.save(dst)
|
226 |
return dst
|
227 |
|
228 |
+
@spaces.GPU
|
229 |
def stage_2_i2v(image, seed, step, save_folder):
|
230 |
if isinstance(image, str):
|
231 |
image = Image.open(image)
|
|
|
263 |
obj_dst = save_folder + '/mesh_vertex_colors.obj' # gradio just only can show vertex shading
|
264 |
return obj_dst, glb_dst
|
265 |
|
266 |
+
@spaces.GPU
|
267 |
def stage_3p_baking(save_folder, color, bake, force, front, others, align_times):
|
268 |
if color == "texture" and bake:
|
269 |
obj_dst = worker_baker(save_folder, force, front, others, align_times)
|
|
|
272 |
else:
|
273 |
return None
|
274 |
|
275 |
+
@spaces.GPU
|
276 |
def stage_4_gif(save_folder, color, bake, render):
|
277 |
if not render: return None
|
278 |
baked_fld_list = sorted(glob(save_folder + '/view_*/bake/mesh.obj'))
|