Update app.py
Browse files
app.py
CHANGED
@@ -119,6 +119,7 @@ def pil_to_binary_mask(pil_image, threshold=0):
|
|
119 |
output_mask = Image.fromarray(mask)
|
120 |
return output_mask
|
121 |
|
|
|
122 |
def start_tryon(dict, garm_img, garment_des, is_checked, is_checked_crop, denoise_steps, seed):
|
123 |
device = "cuda"
|
124 |
openpose_model.preprocessor.body_estimation.model.to(device)
|
|
|
119 |
output_mask = Image.fromarray(mask)
|
120 |
return output_mask
|
121 |
|
122 |
+
@spaces.GPU
|
123 |
def start_tryon(dict, garm_img, garment_des, is_checked, is_checked_crop, denoise_steps, seed):
|
124 |
device = "cuda"
|
125 |
openpose_model.preprocessor.body_estimation.model.to(device)
|