Update app.py
Browse files
app.py
CHANGED
@@ -96,7 +96,7 @@ def resize(img, size):
|
|
96 |
|
97 |
|
98 |
@torch.inference_mode()
|
99 |
-
@spaces.GPU(duration=
|
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,
|