lmzjms commited on
Commit
41d1d11
·
1 Parent(s): 88ae081

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -191,12 +191,14 @@ class ConversationBot:
191
 
192
  if __name__ == '__main__':
193
  bot = ConversationBot({'T2I': 'cuda:0',
 
194
  'T2A': 'cuda:0',
195
  'I2A': 'cuda:0',
196
  'TTS': 'cpu',
197
  'T2S': 'cpu',
198
  'ASR': 'cuda:0',
199
  'A2T': 'cpu',
 
200
  })
201
  with gr.Blocks(css="#chatbot {overflow:auto; height:500px;}") as demo:
202
  with gr.Row():
 
191
 
192
  if __name__ == '__main__':
193
  bot = ConversationBot({'T2I': 'cuda:0',
194
+ 'ImageCaptioning': 'cuda:0',
195
  'T2A': 'cuda:0',
196
  'I2A': 'cuda:0',
197
  'TTS': 'cpu',
198
  'T2S': 'cpu',
199
  'ASR': 'cuda:0',
200
  'A2T': 'cpu',
201
+ 'Inpaint': 'cuda:0'
202
  })
203
  with gr.Blocks(css="#chatbot {overflow:auto; height:500px;}") as demo:
204
  with gr.Row():