Update app.py
Browse files
app.py
CHANGED
@@ -78,10 +78,10 @@ def inference(img, scale, version='v1.3'):
|
|
78 |
cv2.imwrite(save_path, output)
|
79 |
|
80 |
output = cv2.cvtColor(output, cv2.COLOR_BGR2RGB)
|
81 |
-
return output
|
82 |
except Exception as error:
|
83 |
print('global exception', error)
|
84 |
-
return None
|
85 |
|
86 |
|
87 |
title = "GFPGAN"
|
@@ -92,7 +92,6 @@ gr.Interface(
|
|
92 |
gr.inputs.Number(label="Rescaling factor", default=2)
|
93 |
], [
|
94 |
gr.outputs.Image(type="numpy", label="Output (The whole image)"),
|
95 |
-
gr.outputs.File(label="Download the output image")
|
96 |
],
|
97 |
title=title,
|
98 |
examples=[['AI-generate.jpg', 2], ['lincoln.jpg', 2], ['Blake_Lively.jpg', 2],
|
|
|
78 |
cv2.imwrite(save_path, output)
|
79 |
|
80 |
output = cv2.cvtColor(output, cv2.COLOR_BGR2RGB)
|
81 |
+
return output
|
82 |
except Exception as error:
|
83 |
print('global exception', error)
|
84 |
+
return None
|
85 |
|
86 |
|
87 |
title = "GFPGAN"
|
|
|
92 |
gr.inputs.Number(label="Rescaling factor", default=2)
|
93 |
], [
|
94 |
gr.outputs.Image(type="numpy", label="Output (The whole image)"),
|
|
|
95 |
],
|
96 |
title=title,
|
97 |
examples=[['AI-generate.jpg', 2], ['lincoln.jpg', 2], ['Blake_Lively.jpg', 2],
|