Wuvin commited on
Commit
6a0f41b
1 Parent(s): 69ac8ac
Files changed (1) hide show
  1. gradio_app/gradio_3dgen.py +1 -0
gradio_app/gradio_3dgen.py CHANGED
@@ -18,6 +18,7 @@ def run_mv(preview_img, input_processing, seed):
18
  rgb_pils, front_pil = run_mvprediction(preview_img, remove_bg=input_processing, seed=int(seed)) # 6s
19
  return rgb_pils, front_pil
20
 
 
21
  def generate3dv2(preview_img, input_processing, seed, render_video=True, do_refine=True, expansion_weight=0.1, init_type="std"):
22
  if preview_img is None:
23
  raise gr.Error("The input image is none!")
 
18
  rgb_pils, front_pil = run_mvprediction(preview_img, remove_bg=input_processing, seed=int(seed)) # 6s
19
  return rgb_pils, front_pil
20
 
21
+ @spaces.GPU(duration=90) # seems split into multiple part will leads to `RuntimeError`, before fix it, still initialize here
22
  def generate3dv2(preview_img, input_processing, seed, render_video=True, do_refine=True, expansion_weight=0.1, init_type="std"):
23
  if preview_img is None:
24
  raise gr.Error("The input image is none!")