Spaces:
Sleeping
Sleeping
Commit
·
235ca27
1
Parent(s):
7686b0e
Update app.py
Browse files
app.py
CHANGED
|
@@ -20,7 +20,7 @@ upsampler = RealESRGANer(scale=4, model_path=model_path, model=model, tile=0, ti
|
|
| 20 |
model_gfpgan_1_4 = GFPGANer(model_path='GFPGANv1.4.pth', upscale=2, arch='clean', channel_multiplier=2, bg_upsampler=upsampler)
|
| 21 |
|
| 22 |
|
| 23 |
-
|
| 24 |
|
| 25 |
|
| 26 |
# def inference(img, version, scale, weight):
|
|
@@ -63,7 +63,7 @@ def inference(img, version, scale):
|
|
| 63 |
extension = 'png'
|
| 64 |
else:
|
| 65 |
extension = 'jpg'
|
| 66 |
-
save_path = f'out.{extension}'
|
| 67 |
cv2.imwrite(save_path, output)
|
| 68 |
|
| 69 |
output = cv2.cvtColor(output, cv2.COLOR_BGR2RGB)
|
|
|
|
| 20 |
model_gfpgan_1_4 = GFPGANer(model_path='GFPGANv1.4.pth', upscale=2, arch='clean', channel_multiplier=2, bg_upsampler=upsampler)
|
| 21 |
|
| 22 |
|
| 23 |
+
os.makedirs('output', exist_ok=True)
|
| 24 |
|
| 25 |
|
| 26 |
# def inference(img, version, scale, weight):
|
|
|
|
| 63 |
extension = 'png'
|
| 64 |
else:
|
| 65 |
extension = 'jpg'
|
| 66 |
+
save_path = f'output/out.{extension}'
|
| 67 |
cv2.imwrite(save_path, output)
|
| 68 |
|
| 69 |
output = cv2.cvtColor(output, cv2.COLOR_BGR2RGB)
|