jone commited on
Commit
5aaba85
1 Parent(s): 0633da9
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ def inference(img):
46
  cropped_faces, restored_faces, restored_img = restorer.enhance(
47
  input_img, has_aligned=False, only_center_face=False, paste_back=True)
48
 
49
- return Image.fromarray(restored_img[0][:,:,::-1])
50
 
51
  title = "GFP-GAN"
52
  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"
46
  cropped_faces, restored_faces, restored_img = restorer.enhance(
47
  input_img, has_aligned=False, only_center_face=False, paste_back=True)
48
 
49
+ return Image.fromarray(restored_img[:,:,::-1])
50
 
51
  title = "GFP-GAN"
52
  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"