Update app.py

app.py

@@ -248,7 +248,7 @@ def process_image_1(image):
     #cv2.imwrite(f"{rand_im}-vid_tmp_proc.png", image)
     #image = f"{rand_im}-vid_tmp_proc.png"
     image=Image.fromarray(image)
-    out = esr.realesrgan1(img=image, model_name="realesr-general-x4v3", denoise_strength=0.5, face_enhance=False, outscale=1)
+    out = esr.realesrgan1(img=image, model_name="realesr-general-x4v3", denoise_strength=0.5, face_enhance=True, outscale=1)
     #out = os.path.abspath(f"{rand_im}-vid_tmp_proc.png")
     #out_url = f'https://omnibus-vid-url-dl-mod.hf.space/file={out}'
     #out = esr.realesrgan1(image, "realesr-general-x4v3", 0.5, False, 2)
@@ -257,7 +257,7 @@ def process_image_1(image):
     out = np.array(out)
     print (out)
     return out
-def improve_quality():
+def improve_quality(model_name,denoise_strength,outscale,face_enhance):
     
     clip1 = VideoFileClip(f"{uid}-clip.mp4")
     clip = clip1.fl_image(process_image_1)
@@ -292,6 +292,17 @@ with gr.Blocks() as app:
                 hid_box = gr.Textbox(visible=True)
                 hid_fps = gr.Textbox(visible=True)
     with gr.Tab("Quality"):
+        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)        
         impro_btn = gr.Button("Run")
         with gr.Row():
             with gr.Column():
@@ -345,7 +356,7 @@ with gr.Blocks() as app:
 
 
     
-    impro_btn.click(improve_quality,None,clip_out)
+    impro_btn.click(improve_quality,[model_name,denoise_strength,outscale,face_enhance],clip_out)
 
         
     info_btn.click(dl_json,inp_url,info_json)