Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -190,7 +190,7 @@ class ConversationBot:
|
|
| 190 |
|
| 191 |
|
| 192 |
if __name__ == '__main__':
|
| 193 |
-
bot = ConversationBot({'
|
| 194 |
'ImageCaptioning': 'cuda:0',
|
| 195 |
'T2A': 'cuda:0',
|
| 196 |
'I2A': 'cuda:0',
|
|
|
|
| 190 |
|
| 191 |
|
| 192 |
if __name__ == '__main__':
|
| 193 |
+
bot = ConversationBot({'TTS_OOD': 'cuda:0',
|
| 194 |
'ImageCaptioning': 'cuda:0',
|
| 195 |
'T2A': 'cuda:0',
|
| 196 |
'I2A': 'cuda:0',
|