amarax commited on
Commit
8b6c0fe
·
1 Parent(s): b6e2dee

Add image editing and visual question answering back

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -146,8 +146,8 @@ class ConversationBot:
146
 
147
  bot = ConversationBot({'Text2Image': 'cuda:0',
148
  'ImageCaptioning': 'cuda:0',
149
- # 'ImageEditing': 'cuda:0',
150
- # 'VisualQuestionAnswering': 'cuda:0',
151
  # 'Image2Canny': 'cpu',
152
  # 'CannyText2Image': 'cuda:0',
153
  'InstructPix2Pix': 'cuda:0',
@@ -192,13 +192,13 @@ with gr.Blocks(css="#chatbot {overflow:auto; height:500px;}") as demo:
192
  examples=["Generate a figure of a cat running in the garden",
193
  "Replace the cat with a dog",
194
  "Remove the dog in this image",
195
- "Can you detect the canny edge of this image?",
196
- "Can you use this canny image to generate an oil painting of a dog",
197
  "Make it like water-color painting",
198
  "What is the background color",
199
  "Describe this image",
200
- "please detect the depth of this image",
201
- "Can you use this depth image to generate a cute dog",
202
  ],
203
  inputs=txt
204
  )
 
146
 
147
  bot = ConversationBot({'Text2Image': 'cuda:0',
148
  'ImageCaptioning': 'cuda:0',
149
+ 'ImageEditing': 'cuda:0',
150
+ 'VisualQuestionAnswering': 'cuda:0',
151
  # 'Image2Canny': 'cpu',
152
  # 'CannyText2Image': 'cuda:0',
153
  'InstructPix2Pix': 'cuda:0',
 
192
  examples=["Generate a figure of a cat running in the garden",
193
  "Replace the cat with a dog",
194
  "Remove the dog in this image",
195
+ # "Can you detect the canny edge of this image?",
196
+ # "Can you use this canny image to generate an oil painting of a dog",
197
  "Make it like water-color painting",
198
  "What is the background color",
199
  "Describe this image",
200
+ # "please detect the depth of this image",
201
+ # "Can you use this depth image to generate a cute dog",
202
  ],
203
  inputs=txt
204
  )