zxcgqq commited on
Commit
d5ac098
·
1 Parent(s): 5c69c2d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -175,8 +175,8 @@ class ConversationBot:
175
 
176
  return gr.update(visible = True)
177
 
178
- bot = ConversationBot({'Text2Image': 'cuda:0',
179
- 'ImageCaptioning': 'cuda:0'
180
  })
181
 
182
  with gr.Blocks(css="#chatbot {overflow:auto; height:500px;}") as demo:
 
175
 
176
  return gr.update(visible = True)
177
 
178
+ bot = ConversationBot({'Text2Image': 'cpu',
179
+ 'ImageCaptioning': 'cpu'
180
  })
181
 
182
  with gr.Blocks(css="#chatbot {overflow:auto; height:500px;}") as demo: