Update app.py

app.py

@@ -87,9 +87,10 @@ def realesrgan(img, model_name, denoise_strength, face_enhance, outscale, ext):
             channel_multiplier=2,
             bg_upsampler=upsampler)
 
-    try:
+    # parameters
+    if ext == 'png':
         parameters = img.text['parameters']
-    except:
+    else:
         parameters = ''
 
     # Convert the input PIL image to cv2 image, so that it can be processed by realesrgan
@@ -117,7 +118,9 @@ def realesrgan(img, model_name, denoise_strength, face_enhance, outscale, ext):
         metadata = PngInfo()
         if parameters != '':
             metadata.add_text("parameters", parameters)
-        cv2.imwrite(out_filename, output)
+            cv2.imwrite(out_filename, output, [int(cv2.IMWRITE_PNG_COMPRESSION), 0], pnginfo=metadata)
+        else:
+            cv2.imwrite(out_filename, output)
         global last_file
         last_file = out_filename
         return out_filename