youngNdum commited on
Commit
7c05463
·
verified ·
1 Parent(s): b29e228

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +21 -3
app.py CHANGED
@@ -116,13 +116,31 @@ def predict(audio_path):
116
  print(f"Unexpected Exception: {str(e)}")
117
  return None
118
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
119
  demo = gr.Interface(
120
  fn=predict,
121
  inputs=gr.Audio(type="filepath", label="Input Mixture"),
122
  outputs=gr.File(file_count="multiple", label="Separated Stems"),
123
- title="DTTNet Separation API (STFT/ISTFT Patched)",
124
- api_name="predict"
125
- )
 
 
126
 
127
  if __name__ == "__main__":
128
  demo.launch(server_name="0.0.0.0", show_error=True)
 
116
  print(f"Unexpected Exception: {str(e)}")
117
  return None
118
 
119
+ article = """
120
+ ### 📜 Citation & Credits
121
+ This Space is based on the official implementation of:
122
+ **Dual-Path TFC-TDF UNet for Music Source Separation (DTTNet)**
123
+ *Published in ICASSP 2024*
124
+ **Original Paper**: [arXiv:2309.08684](https://arxiv.org/abs/2309.08684)
125
+ **Official GitHub**: [re-searcher/DTTNet](https://github.com/re-searcher/DTTNet)
126
+ @INPROCEEDINGS{chen_dttnet_2024,
127
+ author={Chen, Junyu and Vekkot, Susmitha and Shukla, Pancham},
128
+ booktitle={ICASSP 2024},
129
+ title={Music Source Separation Based on a Lightweight Deep Learning Framework (DTTNET: DUAL-PATH TFC-TDF UNET)},
130
+ year={2024},
131
+ doi={10.1109/ICASSP48485.2024.10448020}}
132
+ """
133
+
134
+
135
  demo = gr.Interface(
136
  fn=predict,
137
  inputs=gr.Audio(type="filepath", label="Input Mixture"),
138
  outputs=gr.File(file_count="multiple", label="Separated Stems"),
139
+ title="DTTNet Separation API (Patched)",
140
+ description="High-performance music source separation based on ICASSP 2024 paper.",
141
+ article=article,
142
+ api_name="predict"
143
+ )
144
 
145
  if __name__ == "__main__":
146
  demo.launch(server_name="0.0.0.0", show_error=True)