roychao19477 commited on
Commit
b753bba
·
1 Parent(s): 9657939
Files changed (1) hide show
  1. app.py +1 -9
app.py CHANGED
@@ -80,18 +80,10 @@ def enhance(filepath):
80
  ax.set_title("Enhanced Spectrogram")
81
  plt.colorbar(format="%+2.0f dB", ax=ax)
82
 
83
- return "enhanced.wav", fig
84
-
85
- # --- Header markdown ---
86
- ABOUT = """
87
- # SEMamba: Speech Enhancement
88
- Unofficial demo for [SEMamba](https://github.com/RoyChao19477/SEMamba), a Mamba-based speech enhancement model from SLT 2024.
89
- Upload or record a noisy audio file to enhance clarity and view the spectrogram.
90
- """
91
 
92
  # --- Build demo ---
93
  with gr.Blocks() as demo:
94
- gr.Markdown(ABOUT)
95
  audio_in = gr.Audio(label="Input Audio", type="filepath", interactive=True)
96
  run_btn = gr.Button("Enhance", variant="primary")
97
  enhanced_audio = gr.Audio(label="Enhanced Audio", type="filepath", interactive=False)
 
80
  ax.set_title("Enhanced Spectrogram")
81
  plt.colorbar(format="%+2.0f dB", ax=ax)
82
 
83
+ return out, fig
 
 
 
 
 
 
 
84
 
85
  # --- Build demo ---
86
  with gr.Blocks() as demo:
 
87
  audio_in = gr.Audio(label="Input Audio", type="filepath", interactive=True)
88
  run_btn = gr.Button("Enhance", variant="primary")
89
  enhanced_audio = gr.Audio(label="Enhanced Audio", type="filepath", interactive=False)