abreza commited on
Commit
8c09611
1 Parent(s): 8e8cc15

Update gradio_app/gradio_3dgen.py

Browse files
Files changed (1) hide show
  1. gradio_app/gradio_3dgen.py +2 -2
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=100)
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=100) # 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!")
 
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!")