chenlei
commited on
Commit
·
79eb473
1
Parent(s):
18d1dd2
update
Browse files- run/gradio_ootd.py +3 -3
run/gradio_ootd.py
CHANGED
|
@@ -21,9 +21,9 @@ openpose_model_hd = OpenPose(0)
|
|
| 21 |
parsing_model_hd = Parsing(0)
|
| 22 |
ootd_model_hd = OOTDiffusionHD(0)
|
| 23 |
|
| 24 |
-
|
| 25 |
-
|
| 26 |
-
|
| 27 |
|
| 28 |
|
| 29 |
category_dict = ['upperbody', 'lowerbody', 'dress']
|
|
|
|
| 21 |
parsing_model_hd = Parsing(0)
|
| 22 |
ootd_model_hd = OOTDiffusionHD(0)
|
| 23 |
|
| 24 |
+
openpose_model_dc = OpenPose(0)
|
| 25 |
+
parsing_model_dc = Parsing(0)
|
| 26 |
+
ootd_model_dc = OOTDiffusionDC(0)
|
| 27 |
|
| 28 |
|
| 29 |
category_dict = ['upperbody', 'lowerbody', 'dress']
|