Update app.py

app.py

@@ -14,7 +14,7 @@ last_file = None
 img_mode = "RGBA"
 
 
-def realesrgan(img, model_name, denoise_strength, face_enhance, outscale):
+def realesrgan(img, model_name, denoise_strength, face_enhance, outscale, ext):
     """Real-ESRGAN function to restore (and upscale) images.
     """
     if not img:
@@ -105,7 +105,7 @@ def realesrgan(img, model_name, denoise_strength, face_enhance, outscale):
             extension = 'jpg'
 
         #out_filename = f"output_{rnd_string(8)}.{extension}"
-        out_filename = f"output_{rnd_string(8)}.png"
+        out_filename = f"output_{rnd_string(8)}." + ext
         cv2.imwrite(out_filename, output)
         global last_file
         last_file = out_filename
@@ -190,6 +190,9 @@ def main():
                                      minimum=1, maximum=10, step=1, value=4, show_label=True)
                 face_enhance = gr.Checkbox(label="Face Enhancement using GFPGAN (Doesn't work for anime images)",
                                            value=False, show_label=True)
+                ext = gr.Dropdown(label="Output Ext",
+                                         choices=["png", "jpg"],
+                                         value="png", show_label=True)
 
         with gr.Row():
             with gr.Group():
@@ -202,7 +205,7 @@ def main():
         # Event listeners:
         input_image.change(fn=image_properties, inputs=input_image, outputs=input_image_properties)
         restore_btn.click(fn=realesrgan,
-                          inputs=[input_image, model_name, denoise_strength, face_enhance, outscale],
+                          inputs=[input_image, model_name, denoise_strength, face_enhance, outscale, ext],
                           outputs=output_image,
                           api_name="upscale")