Update run/gradio_ootd.py
Browse files- 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(
|
|
|
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()
|