Spaces:
Running
on
Zero
Running
on
Zero
bug fix
Browse files
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!")
|