Update app.py

app.py

@@ -26,6 +26,10 @@ def read_main():
 def realesrgan(img, model_name, denoise_strength, face_enhance, outscale):
     """Real-ESRGAN function to restore (and upscale) images.
     """
+    choices=["RealESRGAN_x4plus", "RealESRNet_x4plus", "RealESRGAN_x4plus_anime_6B",
+                                                  "RealESRGAN_x2plus", "realesr-general-x4v3"]
+        model_name = choices[model_name]
+    model_name = choices[model_name]
     if not img:
         return
 
@@ -193,14 +197,15 @@ def main():
 
         with gr.Accordion("Options/Parameters"):
             with gr.Row():
-                model_name = "realesr-general-x4v3",
+                model_name = gr.Slider(label="Model to use",
+                                             minimum=0, maximum=4, step=1, value=4)
                 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")