Update app.py

app.py

@@ -1,15 +1,15 @@
 import gradio as gr
-from speechbrain.pretrained import SepformerSeparation as separator
 import torchaudio
 import torch
 import os
 from pydub import AudioSegment
 import tempfile
+from speechbrain.pretrained.separation import SepformerSeparation
 
 class AudioDenoiser:
     def __init__(self):
         # Initialize the SepFormer model for audio enhancement
-        self.model = separator.from_hparams(
+        self.model = SepformerSeparation.from_hparams(
             source="speechbrain/sepformer-dns4-16k-enhancement",
             savedir='pretrained_models/sepformer-dns4-16k-enhancement'
         )
@@ -98,8 +98,7 @@ def create_gradio_interface():
         fn=denoiser.enhance_audio,
         inputs=gr.Audio(
             type="filepath",
-            label="Upload Noisy Audio",
-            source="upload"
+            label="Upload Noisy Audio"
         ),
         outputs=gr.Audio(
             label="Enhanced Audio",