flamehaze1115 commited on
Commit
350d8f2
1 Parent(s): 693f65b

Update gradio_app.py

Browse files
Files changed (1) hide show
  1. gradio_app.py +2 -4
gradio_app.py CHANGED
@@ -47,6 +47,7 @@ _TITLE = '''Wonder3D: Single Image to 3D using Cross-Domain Diffusion'''
47
  _DESCRIPTION = '''
48
  <div>
49
  Generate consistent multi-view normals maps and color images.
 
50
  </div>
51
  '''
52
  _GPU_ID = 0
@@ -332,10 +333,7 @@ def run_demo():
332
  first_stage = run_btn.click(fn=partial(preprocess, predictor),
333
  inputs=[input_image, input_processing],
334
  outputs=[processed_image_highres, processed_image], queue=True
335
- )
336
-
337
-
338
- first_stage.success(fn=partial(run_pipeline, pipeline, cfg),
339
  inputs=[processed_image_highres, scale_slider, steps_slider, seed, crop_size],
340
  outputs=[view_1, view_2, view_3, view_4, view_5, view_6, normal_1, normal_2, normal_3, normal_4, normal_5, normal_6]
341
  )
 
47
  _DESCRIPTION = '''
48
  <div>
49
  Generate consistent multi-view normals maps and color images.
50
+ <a style="display:inline-block; margin-left: .5em" href='https://github.com/xxlong0/Wonder3D/'><img src='https://img.shields.io/github/stars/xxlong0/Wonder3D?style=social' /></a>
51
  </div>
52
  '''
53
  _GPU_ID = 0
 
333
  first_stage = run_btn.click(fn=partial(preprocess, predictor),
334
  inputs=[input_image, input_processing],
335
  outputs=[processed_image_highres, processed_image], queue=True
336
+ ).success(fn=partial(run_pipeline, pipeline, cfg),
 
 
 
337
  inputs=[processed_image_highres, scale_slider, steps_slider, seed, crop_size],
338
  outputs=[view_1, view_2, view_3, view_4, view_5, view_6, normal_1, normal_2, normal_3, normal_4, normal_5, normal_6]
339
  )