Ahsen Khaliq commited on
Commit
d558166
1 Parent(s): 83af908

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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"