Update run/gradio_ootd.py
Browse files- run/gradio_ootd.py +1 -1
run/gradio_ootd.py
CHANGED
|
@@ -36,7 +36,7 @@ garment_hd = os.path.join(example_path, 'garment/03244_00.jpg')
|
|
| 36 |
model_dc = os.path.join(example_path, 'model/model_8.png')
|
| 37 |
garment_dc = os.path.join(example_path, 'garment/048554_1.jpg')
|
| 38 |
|
| 39 |
-
|
| 40 |
def process_hd(vton_img, garm_img, n_samples, n_steps, image_scale, seed):
|
| 41 |
model_type = 'hd'
|
| 42 |
category = 0 # 0:upperbody; 1:lowerbody; 2:dress
|
|
|
|
| 36 |
model_dc = os.path.join(example_path, 'model/model_8.png')
|
| 37 |
garment_dc = os.path.join(example_path, 'garment/048554_1.jpg')
|
| 38 |
|
| 39 |
+
@spaces.GPU
|
| 40 |
def process_hd(vton_img, garm_img, n_samples, n_steps, image_scale, seed):
|
| 41 |
model_type = 'hd'
|
| 42 |
category = 0 # 0:upperbody; 1:lowerbody; 2:dress
|