Spaces:
Sleeping
Sleeping
jsaplication
commited on
Commit
•
7686b0e
1
Parent(s):
f8fca07
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 |
-
os.makedirs('output', exist_ok=True)
|
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'
|
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'out.{extension}'
|
67 |
cv2.imwrite(save_path, output)
|
68 |
|
69 |
output = cv2.cvtColor(output, cv2.COLOR_BGR2RGB)
|