Spaces:
Running
Running
Ahsen Khaliq
commited on
Commit
·
d558166
1
Parent(s):
83af908
Update app.py
Browse files
app.py
CHANGED
@@ -117,8 +117,7 @@ faceenhancer = FaceEnhancement(size=size, model=model, use_sr=use_sr, sr_model=s
|
|
117 |
def inference(file):
|
118 |
im = cv2.imread(file, cv2.IMREAD_COLOR)
|
119 |
img, orig_faces, enhanced_faces = faceenhancer.process(im)
|
120 |
-
|
121 |
-
return enhanced_faces[0]
|
122 |
|
123 |
title = "GFP-GAN"
|
124 |
description = "Gradio demo for GFP-GAN: Towards Real-World Blind Face Restoration with Generative Facial Prior. To use it, simply upload your image, or click one of the examples to load them. Read more at the links below. Please click submit only once"
|
|
|
117 |
def inference(file):
|
118 |
im = cv2.imread(file, cv2.IMREAD_COLOR)
|
119 |
img, orig_faces, enhanced_faces = faceenhancer.process(im)
|
120 |
+
return enhanced_faces[0][:,:,::-1]
|
|
|
121 |
|
122 |
title = "GFP-GAN"
|
123 |
description = "Gradio demo for GFP-GAN: Towards Real-World Blind Face Restoration with Generative Facial Prior. To use it, simply upload your image, or click one of the examples to load them. Read more at the links below. Please click submit only once"
|