Hugging Face's logo

Spaces:
zhengchong
/
CatVTON
Running on Zero

App Files Community
2
zhengchong commited on
Commit
640d658
•
1 Parent(s): 6927e60

feat: Add GPU duration parameter to submit_function

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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,