Spaces:
Runtime error
Runtime error
Commit
·
640d658
1
Parent(s):
6927e60
feat: Add GPU duration parameter to submit_function
Browse files
app.py
CHANGED
|
@@ -123,7 +123,7 @@ automasker = AutoMasker(
|
|
| 123 |
device='cuda',
|
| 124 |
)
|
| 125 |
|
| 126 |
-
@spaces.GPU
|
| 127 |
def submit_function(
|
| 128 |
person_image,
|
| 129 |
cloth_image,
|
|
|
|
| 123 |
device='cuda',
|
| 124 |
)
|
| 125 |
|
| 126 |
+
@spaces.GPU(duration=120)
|
| 127 |
def submit_function(
|
| 128 |
person_image,
|
| 129 |
cloth_image,
|