Spaces:
Runtime error
Runtime error
Update gradio_app/gradio_3dgen.py
Browse files
gradio_app/gradio_3dgen.py
CHANGED
@@ -11,14 +11,14 @@ from scripts.utils import save_obj_and_video
|
|
11 |
# from scripts.multiview_inference import geo_reconstruct
|
12 |
from scripts.multiview_inference import geo_reconstruct_part1, geo_reconstruct_part2, geo_reconstruct_part3
|
13 |
|
14 |
-
@spaces.GPU(duration=
|
15 |
def run_mv(preview_img, input_processing, seed):
|
16 |
if preview_img.size[0] <= 512:
|
17 |
preview_img = run_sr_fast([preview_img])[0]
|
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=
|
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!")
|
|
|
11 |
# from scripts.multiview_inference import geo_reconstruct
|
12 |
from scripts.multiview_inference import geo_reconstruct_part1, geo_reconstruct_part2, geo_reconstruct_part3
|
13 |
|
14 |
+
@spaces.GPU(duration=200)
|
15 |
def run_mv(preview_img, input_processing, seed):
|
16 |
if preview_img.size[0] <= 512:
|
17 |
preview_img = run_sr_fast([preview_img])[0]
|
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=200) # 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!")
|