Update app.py

app.py

@@ -1,16 +1,16 @@
 import gradio as gr
-import torch
+from speechbrain.pretrained import MetricGAN
 
-# Load the model
-model = torch.hub.load("speechbrain/metricgan-plus-voicebank", "enhance")
+# Load the MetricGAN model from Hugging Face
+model = MetricGAN.from_hparams(source="speechbrain/metricgan-plus-voicebank")
 
-# Define the function to pass audio through the model
-def enhance_speech(audio):
-    # Process the audio file through the loaded model
-    enhanced_audio = model(audio)
+# 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
 
-# Create Gradio Interface
+# Set up the Gradio interface
 iface = gr.Interface(
     fn=enhance_speech,
     inputs=gr.Audio(source="upload", type="filepath"),
@@ -18,5 +18,5 @@ iface = gr.Interface(
     title="Speech Enhancement"
 )
 
-# Launch the interface
+# Launch the Gradio interface
 iface.launch()