Update app.py

app.py

@@ -169,58 +169,54 @@ def image_properties(img):
         return properties
 
 
-def main():
-    # Gradio Interface
-    with gr.Blocks(title="Real-ESRGAN Gradio Demo", theme="dark") as demo:
-
-        gr.Markdown(
-            """# <div align="center"> Real-ESRGAN Demo for Image Restoration and Upscaling </div>
-        <div align="center"><img width="200" height="74" src="https://github.com/xinntao/Real-ESRGAN/raw/master/assets/realesrgan_logo.png"></div>  
-
-        This Gradio Demo was built as my Final Project for **CS50's Introduction to Programming with Python**.  
-        Please visit the [Real-ESRGAN GitHub page](https://github.com/xinntao/Real-ESRGAN) for detailed information about the project.
-        """
-        )
-
-        with gr.Accordion("Options/Parameters"):
-            with gr.Row():
-                model_name = gr.Dropdown(label="Real-ESRGAN inference model to be used",
-                                         choices=["RealESRGAN_x4plus", "RealESRNet_x4plus", "RealESRGAN_x4plus_anime_6B",
-                                                  "RealESRGAN_x2plus", "realesr-general-x4v3"],
-                                         value="realesr-general-x4v3", show_label=True)
-                denoise_strength = gr.Slider(label="Denoise Strength (Used only with the realesr-general-x4v3 model)",
-                                             minimum=0, maximum=1, step=0.1, value=0.5)
-                outscale = gr.Slider(label="Image Upscaling Factor",
-                                     minimum=1, maximum=10, step=1, value=2, show_label=True)
-                face_enhance = gr.Checkbox(label="Face Enhancement using GFPGAN (Doesn't work for anime images)",
-                                           value=False, show_label=True)
+# Gradio Interface
+with gr.Blocks(title="Real-ESRGAN Gradio Demo", theme="dark") as demo:
 
-        with gr.Row():
-            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)
-            output_image = gr.Image(label="Restored Image", image_mode="RGBA")
-        with gr.Row():
-            restore_btn = gr.Button("Restore Image")
-            reset_btn = gr.Button("Reset")
+    gr.Markdown(
+        """# <div align="center"> Real-ESRGAN Demo for Image Restoration and Upscaling </div>
+    <div align="center"><img width="200" height="74" src="https://github.com/xinntao/Real-ESRGAN/raw/master/assets/realesrgan_logo.png"></div>  
 
-        # 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],
-                          outputs=output_image)
-        reset_btn.click(fn=reset, inputs=[], outputs=[output_image, input_image])
-        # reset_btn.click(None, inputs=[], outputs=[input_image], _js="() => (null)\n")
-        # Undocumented method to clear a component's value using Javascript
-
-        gr.Markdown(
-            """*Please note that support for animated GIFs is not yet implemented. Should an animated GIF is chosen for restoration, 
-            the demo will output only the first frame saved in PNG format (to preserve probable transparency).*
-        """
-        )
+    This Gradio Demo was built as my Final Project for **CS50's Introduction to Programming with Python**.  
+    Please visit the [Real-ESRGAN GitHub page](https://github.com/xinntao/Real-ESRGAN) for detailed information about the project.
+    """
+    )
 
-    demo.launch()
+    with gr.Accordion("Options/Parameters"):
+        with gr.Row():
+            model_name = gr.Dropdown(label="Real-ESRGAN inference model to be used",
+                                     choices=["RealESRGAN_x4plus", "RealESRNet_x4plus", "RealESRGAN_x4plus_anime_6B",
+                                              "RealESRGAN_x2plus", "realesr-general-x4v3"],
+                                     value="realesr-general-x4v3", show_label=True)
+            denoise_strength = gr.Slider(label="Denoise Strength (Used only with the realesr-general-x4v3 model)",
+                                         minimum=0, maximum=1, step=0.1, value=0.5)
+            outscale = gr.Slider(label="Image Upscaling Factor",
+                                 minimum=1, maximum=10, step=1, value=2, show_label=True)
+            face_enhance = gr.Checkbox(label="Face Enhancement using GFPGAN (Doesn't work for anime images)",
+                                       value=False, show_label=True)
+
+    with gr.Row():
+        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)
+        output_image = gr.Image(label="Restored Image", image_mode="RGBA")
+    with gr.Row():
+        restore_btn = gr.Button("Restore Image")
+        reset_btn = gr.Button("Reset")
+
+    # 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],
+                      outputs=output_image)
+    reset_btn.click(fn=reset, inputs=[], outputs=[output_image, input_image])
+    # reset_btn.click(None, inputs=[], outputs=[input_image], _js="() => (null)\n")
+    # Undocumented method to clear a component's value using Javascript
+
+    gr.Markdown(
+        """*Please note that support for animated GIFs is not yet implemented. Should an animated GIF is chosen for restoration, 
+        the demo will output only the first frame saved in PNG format (to preserve probable transparency).*
+    """
+    )
 
+demo.launch()
 
-if __name__ == "__main__":
-    main()