Update app.py

app.py

@@ -1,7 +1,22 @@
 import gradio as gr
+import torch
 
-# Load the model without the examples argument
-iface = gr.Interface.load("models/speechbrain/metricgan-plus-voicebank")
+# Load the model
+model = torch.hub.load("speechbrain/metricgan-plus-voicebank", "enhance")
+
+# 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)
+    return enhanced_audio
+
+# Create 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 interface
 iface.launch()