Update app.py

app.py

@@ -9,9 +9,6 @@ from basicsr.utils.download_util import load_file_from_url
 from realesrgan import RealESRGANer
 from realesrgan.archs.srvgg_arch import SRVGGNetCompact
 
-from PIL import Image
-from PIL.PngImagePlugin import PngInfo
-
 
 last_file = None
 img_mode = "RGBA"
@@ -87,12 +84,6 @@ def realesrgan(img, model_name, denoise_strength, face_enhance, outscale, ext):
             channel_multiplier=2,
             bg_upsampler=upsampler)
 
-    # parameters
-    if ext == 'png':
-        parameters = img.text['parameters']
-    else:
-        parameters = ''
-
     # Convert the input PIL image to cv2 image, so that it can be processed by realesrgan
     cv_img = numpy.array(img)
     img = cv2.cvtColor(cv_img, cv2.COLOR_RGBA2BGRA)
@@ -115,11 +106,6 @@ def realesrgan(img, model_name, denoise_strength, face_enhance, outscale, ext):
 
         #out_filename = f"output_{rnd_string(8)}.{extension}"
         out_filename = f"output_{rnd_string(8)}." + ext
-        #if parameters != '':
-        #    metadata = PngInfo()
-        #    metadata.add_text("parameters", parameters)
-        #    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
@@ -183,9 +169,6 @@ def image_properties(img):
         properties = f"Width: {img.size[0]}, Height: {img.size[1]}  |  Color Mode: {img_mode}"
         return properties
 
-def image_metadata(img):
-    return img.text['parameters']
-
 
 def main():
     # Gradio Interface
@@ -215,7 +198,6 @@ def main():
             with gr.Group():
                 input_image = gr.Image(label="Source Image", type="pil", image_mode="RGBA")
                 input_image_properties = gr.Textbox(label="Image Properties", max_lines=1)
-                #input_image_metadata = gr.Textbox(label="Image Metadata", max_lines=1)
             output_image = gr.Image(label="Restored Image", image_mode="RGBA", show_share_button=False)
         with gr.Row():
             restore_btn = gr.Button("Upscale")