lmzjms commited on
Commit
fe9952a
·
1 Parent(s): bc74c9d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -430,13 +430,13 @@ class ConversationBot:
430
 
431
  if __name__ == '__main__':
432
  bot = ConversationBot({'ImageCaptioning': 'cuda:0',
433
- 'T2A': 'cuda:0',
434
- 'I2A': 'cuda:0',
435
  'TTS': 'cpu',
436
  'T2S': 'cpu',
437
  'ASR': 'cuda:0',
438
  'A2T': 'cpu',
439
- 'Inpaint': 'cuda:0',
440
  'SoundDetection': 'cpu',
441
  'Binaural': 'cuda:0',
442
  'SoundExtraction': 'cuda:0',
 
430
 
431
  if __name__ == '__main__':
432
  bot = ConversationBot({'ImageCaptioning': 'cuda:0',
433
+ # 'T2A': 'cuda:0',
434
+ # 'I2A': 'cuda:0',
435
  'TTS': 'cpu',
436
  'T2S': 'cpu',
437
  'ASR': 'cuda:0',
438
  'A2T': 'cpu',
439
+ # 'Inpaint': 'cuda:0',
440
  'SoundDetection': 'cpu',
441
  'Binaural': 'cuda:0',
442
  'SoundExtraction': 'cuda:0',