Spaces:

microsoft
/

visual_chatgpt

Runtime error

LanHarmony commited on Apr 17, 2023

Commit

f97efd7

1 Parent(s): 8c482b3

support GroundingDINO and segment-anything

Files changed (1) hide show

app.py CHANGED Viewed

@@ -205,7 +205,10 @@ class ConversationBot:
         return gr.update(visible = True)
-bot = ConversationBot({'Text2Image': 'cuda:0',
                        'ImageCaptioning': 'cuda:0',
                        'ImageEditing': 'cuda:0',
                        'VisualQuestionAnswering': 'cuda:0',

         return gr.update(visible = True)
+bot = ConversationBot({'Text2Box': 'cuda:0',
+                       'Segmenting': 'cuda:0',
+                       'Inpainting': 'cuda:0',
+                       'Text2Image': 'cuda:0',
                        'ImageCaptioning': 'cuda:0',
                        'ImageEditing': 'cuda:0',
                        'VisualQuestionAnswering': 'cuda:0',