levihsu commited on
Commit
83ed56c
1 Parent(s): 470ba38

Update run/gradio_ootd.py

Browse files
Files changed (1) hide show
  1. run/gradio_ootd.py +3 -1
run/gradio_ootd.py CHANGED
@@ -15,6 +15,8 @@ from preprocess.humanparsing.run_parsing import Parsing
15
  from ootd.inference_ootd_hd import OOTDiffusionHD
16
  from ootd.inference_ootd_dc import OOTDiffusionDC
17
 
 
 
18
 
19
  openpose_model_hd = OpenPose(0)
20
  parsing_model_hd = Parsing(0)
@@ -256,4 +258,4 @@ with block:
256
  ips_dc = [vton_img_dc, garm_img_dc, category_dc, n_samples_dc, n_steps_dc, image_scale_dc, seed_dc]
257
  run_button_dc.click(fn=process_dc, inputs=ips_dc, outputs=[result_gallery_dc])
258
 
259
- block.launch(server_name='0.0.0.0', server_port=7865)
 
15
  from ootd.inference_ootd_hd import OOTDiffusionHD
16
  from ootd.inference_ootd_dc import OOTDiffusionDC
17
 
18
+ import spaces
19
+
20
 
21
  openpose_model_hd = OpenPose(0)
22
  parsing_model_hd = Parsing(0)
 
258
  ips_dc = [vton_img_dc, garm_img_dc, category_dc, n_samples_dc, n_steps_dc, image_scale_dc, seed_dc]
259
  run_button_dc.click(fn=process_dc, inputs=ips_dc, outputs=[result_gallery_dc])
260
 
261
+ block.launch()