Hunsain Mazhar commited on
Commit ·
27c04bc
1
Parent(s): c3bbba4
Reduce GPU duration for inference from 180 to 120 seconds
Browse files
app.py
CHANGED
|
@@ -157,7 +157,7 @@ tensor_transform = transforms.Compose([transforms.ToTensor(), transforms.Normali
|
|
| 157 |
# ---------------------------------------------------------
|
| 158 |
# 4. INFERENCE
|
| 159 |
# ---------------------------------------------------------
|
| 160 |
-
@spaces.GPU(duration=
|
| 161 |
def start_tryon(human_img, garm_img, garment_des, is_checked, is_checked_crop, denoise_steps, seed):
|
| 162 |
device = "cuda"
|
| 163 |
|
|
|
|
| 157 |
# ---------------------------------------------------------
|
| 158 |
# 4. INFERENCE
|
| 159 |
# ---------------------------------------------------------
|
| 160 |
+
@spaces.GPU(duration=120)
|
| 161 |
def start_tryon(human_img, garm_img, garment_des, is_checked, is_checked_crop, denoise_steps, seed):
|
| 162 |
device = "cuda"
|
| 163 |
|