Spaces:
doevent
/
Running on Zero

doevent commited on
Commit
48c113a
·
verified ·
1 Parent(s): f45df51

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -96,7 +96,7 @@ def resize(img, size):
96
 
97
 
98
  @torch.inference_mode()
99
- @spaces.GPU(duration=30)
100
  def enhance_face(img, face_helper, has_aligned, num_flow_steps, scale=2):
101
  face_helper.clean_all()
102
  if has_aligned: # The inputs are already aligned
@@ -146,7 +146,7 @@ def enhance_face(img, face_helper, has_aligned, num_flow_steps, scale=2):
146
 
147
 
148
  @torch.inference_mode()
149
- @spaces.GPU()
150
  def inference(
151
  img,
152
  randomize_seed,
 
96
 
97
 
98
  @torch.inference_mode()
99
+ @spaces.GPU(duration=20)
100
  def enhance_face(img, face_helper, has_aligned, num_flow_steps, scale=2):
101
  face_helper.clean_all()
102
  if has_aligned: # The inputs are already aligned
 
146
 
147
 
148
  @torch.inference_mode()
149
+ @spaces.GPU(duration=20)
150
  def inference(
151
  img,
152
  randomize_seed,