Update app.py

app.py

@@ -119,7 +119,12 @@ def inference(img, version, scale, blur_face):
         cv2.imwrite(save_path, output)
 
         output = cv2.cvtColor(output, cv2.COLOR_BGR2RGB)
-        return output, save_path, [cv2.cvtColor(e, cv2.COLOR_BGR2RGB) for e in face_enhancer.face_helper.org_cropped_faces]
+        return (
+            output,
+            save_path,
+            [cv2.cvtColor(e, cv2.COLOR_BGR2RGB) for e in face_enhancer.face_helper.org_cropped_faces],
+            [cv2.cvtColor(e, cv2.COLOR_BGR2RGB) for e in face_enhancer.face_helper.restored_faces]
+        )
     except Exception as error:
         print('global exception', error)
         return None, None
@@ -153,7 +158,8 @@ demo = gr.Interface(
     ], [
         gr.outputs.Image(type="numpy", label="Output (The whole image)"),
         gr.outputs.File(label="Download the output image"),
-        gr.Gallery(label="All faces").style(grid=[2], height="auto")
+        gr.Gallery(label="Input faces").style(grid=[2], height="auto"),
+        gr.Gallery(label="Output faces").style(grid=[2], height="auto")
     ],
     title=title,
     description=description,