Update app.py

app.py

@@ -2,12 +2,12 @@ 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")
+model = MetricGAN.from_hparams(source="speechbrain/metricgan-plus-voicebank", savedir="tmpdir_metricgan")
 
 # Define a function to enhance speech
-def enhance_speech(audio_path):
+def enhance_speech(audio):
     # Process the uploaded audio file through the model
-    enhanced_audio = model.enhance(audio_path)
+    enhanced_audio = model.enhance_file(audio)
     return enhanced_audio
 
 # Set up the Gradio interface
@@ -15,7 +15,8 @@ iface = gr.Interface(
     fn=enhance_speech,
     inputs=gr.Audio(source="upload", type="filepath"),
     outputs=gr.Audio(type="filepath"),
-    title="Speech Enhancement"
+    title="Speech Enhancement",
+    description="Upload a noisy audio file to enhance it using MetricGAN."
 )
 
 # Launch the Gradio interface