Update gfpgan/utils.py

gfpgan/utils.py

@@ -109,9 +109,9 @@ class GFPGANer():
             self.face_helper.get_inverse_affine(None)
             # paste each restored face to the input image
             restored_img = self.face_helper.paste_faces_to_input_image(upsample_img=bg_img)
-            return self.face_helper.cropped_faces, restored_face, restored_img
+            return self.face_helper.cropped_faces, self.face_helper.restored_faces, restored_img
         else:
-            return self.face_helper.cropped_faces, restored_face, None
+            return self.face_helper.cropped_faces, self.face_helper.restored_faces, None
 
 
 def load_file_from_url(url, model_dir=None, progress=True, file_name=None):