Spaces:
Build error
Build error
Update app.py
Browse files
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():
|