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():
|