Huiwenshi commited on
Commit
a3073d2
·
verified ·
1 Parent(s): 63c61eb

Upload app_hg.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app_hg.py +4 -4
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
- @space.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,7 +225,7 @@ def stage_1_xbg(image, save_folder, force_remove):
225
  rgba.save(dst)
226
  return dst
227
 
228
- @space.GPU
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
- @space.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,7 +272,7 @@ def stage_3p_baking(save_folder, color, bake, force, front, others, align_times)
272
  else:
273
  return None
274
 
275
- @space.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'))
 
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'))