Spaces:

microsoft
/

visual_chatgpt

Runtime error

LanHarmony commited on Mar 13, 2023

Commit

a009035

•

1 Parent(s): ee54eed

api key

Files changed (1) hide show

app.py CHANGED Viewed

@@ -136,7 +136,13 @@ class ConversationBot:
         return gr.update(visible = True)
-bot = ConversationBot({'Text2Image':'cuda:0', 'ImageCaptioning':'cuda:0',})
 with gr.Blocks(css="#chatbot .overflow-y-auto{height:500px}") as demo:
     with gr.Row():

         return gr.update(visible = True)
+bot = ConversationBot({'Text2Image':'cuda:0',
+                       'ImageCaptioning':'cuda:0',
+                       'ImageEditing': 'cuda:0',
+                       'VisualQuestionAnswering': 'cuda:0',
+                       'Image2Canny':'cpu',
+                       'CannyText2Image':'cuda:0',
+                       'InstructPix2Pix':'cuda:0'})
 with gr.Blocks(css="#chatbot .overflow-y-auto{height:500px}") as demo:
     with gr.Row():