Spaces:
Paused
Paused
Add image editing and visual question answering back
Browse files
app.py
CHANGED
@@ -146,8 +146,8 @@ class ConversationBot:
|
|
146 |
|
147 |
bot = ConversationBot({'Text2Image': 'cuda:0',
|
148 |
'ImageCaptioning': 'cuda:0',
|
149 |
-
|
150 |
-
|
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 |
-
|
196 |
-
|
197 |
"Make it like water-color painting",
|
198 |
"What is the background color",
|
199 |
"Describe this image",
|
200 |
-
|
201 |
-
|
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 |
)
|