| import gradio as gr | |
| from speechbrain.pretrained import MetricGAN | |
| # Load the MetricGAN model from Hugging Face | |
| model = MetricGAN.from_hparams(source="speechbrain/metricgan-plus-voicebank") | |
| # Define a function to enhance speech | |
| def enhance_speech(audio_path): | |
| # Process the uploaded audio file through the model | |
| enhanced_audio = model.enhance(audio_path) | |
| return enhanced_audio | |
| # Set up the Gradio interface | |
| iface = gr.Interface( | |
| fn=enhance_speech, | |
| inputs=gr.Audio(source="upload", type="filepath"), | |
| outputs=gr.Audio(type="filepath"), | |
| title="Speech Enhancement" | |
| ) | |
| # Launch the Gradio interface | |
| iface.launch() | |