Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -41,6 +41,8 @@ class ChatBot:
|
|
41 |
def draw_boxes(self, response):
|
42 |
image = self.tokenizer.draw_bbox_on_latest_picture(response, self.history)
|
43 |
if image is not None:
|
|
|
|
|
44 |
buffered = io.BytesIO()
|
45 |
image.save(buffered, format="PNG")
|
46 |
img_str = base64.b64encode(buffered.getvalue()).decode()
|
|
|
41 |
def draw_boxes(self, response):
|
42 |
image = self.tokenizer.draw_bbox_on_latest_picture(response, self.history)
|
43 |
if image is not None:
|
44 |
+
if not isinstance(image, PILImage.Image):
|
45 |
+
image = PILImage.fromarray(image.get_image())
|
46 |
buffered = io.BytesIO()
|
47 |
image.save(buffered, format="PNG")
|
48 |
img_str = base64.b64encode(buffered.getvalue()).decode()
|