lmzjms commited on
Commit
810d95e
1 Parent(s): 329cb64

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -216,7 +216,7 @@ if __name__ == '__main__':
216
  'SoundDetection': 'cpu',
217
  'Binaural': 'cuda:0',
218
  'SoundExtraction': 'cuda:0',
219
- 'TargetSoundDetection': 'cpu'
220
  })
221
  with gr.Blocks(css="#chatbot {overflow:auto; height:500px;}") as demo:
222
  gr.Markdown(_DESCRIPTION)
 
216
  'SoundDetection': 'cpu',
217
  'Binaural': 'cuda:0',
218
  'SoundExtraction': 'cuda:0',
219
+ 'TargetSoundDetection': 'cuda:0'
220
  })
221
  with gr.Blocks(css="#chatbot {overflow:auto; height:500px;}") as demo:
222
  gr.Markdown(_DESCRIPTION)