Update app.py

app.py

@@ -3,7 +3,6 @@ import torch
 import torchaudio
 from speechbrain.inference.enhancement import SpectralMaskEnhancement
 from speechbrain.inference.separation import SepformerSeparation as separator
-import torchaudio
 
 # Load the enhancement model
 model = separator.from_hparams(
@@ -14,10 +13,10 @@ model = separator.from_hparams(
 # Define the enhancement function
 def enhance_audio(noisy_audio):
     # Load and add a batch dimension to the audio tensor
-    noisy = enhance_model.load_audio(noisy_audio).unsqueeze(0)
+    noisy = model.load_audio(noisy_audio).unsqueeze(0)
 
     # Enhance the audio
-    enhanced = enhance_model.enhance_batch(noisy, lengths=torch.tensor([1.0]))
+    enhanced = model.enhance_batch(noisy, lengths=torch.tensor([1.0]))
 
     # Save enhanced audio to a temporary file
     enhanced_path = "enhanced.wav"
@@ -34,5 +33,5 @@ interface = gr.Interface(
     description="Upload a noisy audio file to enhance the quality. The enhanced audio can be downloaded after processing."
 )
 
-# Launch the Gradio app
-interface.launch()
+# Launch the Gradio app with public link enabled
+interface.launch(share=True)