Update app.py

app.py

@@ -41,6 +41,8 @@ class ChatBot:
     def draw_boxes(self, response):
         image = self.tokenizer.draw_bbox_on_latest_picture(response, self.history)
         if image is not None:
+            if not isinstance(image, PILImage.Image):
+                image = PILImage.fromarray(image.get_image())
             buffered = io.BytesIO()
             image.save(buffered, format="PNG")
             img_str = base64.b64encode(buffered.getvalue()).decode()