app.py

@@ -20,11 +20,10 @@ def inference(file, mode):
 
     if mode == "enhance":
         model = {'name':'GPEN-512', 'size':512}
-        im = cv2.imread(file, cv2.IMREAD_COLOR) 
-        im = cv2.resize(im, (0,0), fx=2, fy=2)
+        im = cv2.imread(file, cv2.IMREAD_COLOR)
         faceenhancer = FaceEnhancement(size=model['size'], model=model['name'], channel_multiplier=2, device='cpu')
         img, orig_faces, enhanced_faces = faceenhancer.process(im)
-        return img[0][:,:,::-1]
+        return img
     else:
         model = {'name':'GPEN-1024-Color', 'size':1024}
         grayf = cv2.imread(file, cv2.IMREAD_GRAYSCALE)
@@ -44,7 +43,7 @@ article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2105.060
 gr.Interface(
     inference, 
     [gr.inputs.Image(type="filepath", label="Input"),gr.inputs.Radio(["enhance","colorize"], type="value", default="enhance", label="model type")], 
-    gr.outputs.Image(type="file", label="Output"),
+    gr.outputs.Image(type="numpy", label="Output"),
     title=title,
     description=description,
     article=article,