Update app.py

app.py

@@ -1,8 +1,8 @@
 import gradio as gr
-from speechbrain.pretrained import Tacotron2, HIFIGAN  # Adjust imports for your use case
+from speechbrain.inference import SpectralMaskGAN
 
-# Load the MetricGAN model using Pretrained
-model = HIFIGAN.from_hparams(source="speechbrain/metricgan-plus-voicebank", savedir="tmpdir_metricgan")
+# Load the MetricGAN model
+model = SpectralMaskGAN.from_hparams(source="speechbrain/metricgan-plus-voicebank", savedir="tmpdir_metricgan")
 
 # Define a function to enhance speech
 def enhance_speech(audio):
@@ -21,4 +21,3 @@ iface = gr.Interface(
 
 # Launch the Gradio interface
 iface.launch()
-