Tonic commited on
Commit
c7140ed
·
1 Parent(s): 002bc1f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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()