Update app.py

app.py

@@ -12,8 +12,6 @@ torch.hub.download_url_to_file('https://upload.wikimedia.org/wikipedia/commons/5
 import cv2
 import glob
 import numpy as np
-import os
-import torch
 from basicsr.utils import imwrite
 from gfpgan import GFPGANer
 
@@ -51,16 +49,10 @@ restorer = GFPGANer(
 
 def inference(img):
     input_img = cv2.imread(img, cv2.IMREAD_COLOR)
-    # restore faces and background if necessary
     cropped_faces, restored_faces, restored_img = restorer.enhance(
         input_img, has_aligned=False, only_center_face=False, paste_back=True)
-    # save faces
-    # save cropped face
-   
-    # save comparison image
-    cmp_img = np.concatenate((cropped_faces[0], restored_faces[0]), axis=1)
-   
-    return Image.fromarray(cmp_img)
+    
+    return Image.fromarray(restored_faces[0])
         
 title = "GFP-GAN"
 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"