Update app.py

app.py

@@ -28,7 +28,7 @@ def audio_postprocess(self, y):
 
 gr.Audio.postprocess = audio_postprocess
 def create_vc_fn(model, sid):
-    def vc_fn(input_audio, vc_transform, auto_f0):
+    def vc_fn(input_audio, vc_transform, auto_f0, fmp):
         if input_audio is None:
             return "You need to upload an audio", None
         sampling_rate, audio = input_audio
@@ -44,7 +44,7 @@ def create_vc_fn(model, sid):
         soundfile.write(raw_path, audio, 16000, format="wav")
         raw_path.seek(0)
         out_audio, out_sr = model.infer(sid, vc_transform, raw_path,
-                                       auto_predict_f0=auto_f0,
+                                       auto_predict_f0=auto_f0, f0_mean_pooling=fmp
                                        )
         return "Success", (44100, out_audio.cpu().numpy())
     return vc_fn
@@ -84,12 +84,13 @@ if __name__ == '__main__':
                             vc_input = gr.Audio(label="Input audio"+' (less than 20 seconds)' if limitation else '')
                             vc_transform = gr.Number(label="vc_transform", value=0)
                             auto_f0 = gr.Checkbox(label="auto_f0", value=False)
+                            fmp = gr.Checkbox(label="fmp", value=False)
                             vc_submit = gr.Button("Generate", variant="primary")
 
                         with gr.Column():
                             vc_output1 = gr.Textbox(label="Output Message")
                             vc_output2 = gr.Audio(label="Output Audio")
-                vc_submit.click(vc_fn, [vc_input, vc_transform, auto_f0], [vc_output1, vc_output2])
+                vc_submit.click(vc_fn, [vc_input, vc_transform, auto_f0, fmp], [vc_output1, vc_output2])
             
             """
             for category, link in others.items():