Update app.py
Browse files
app.py
CHANGED
@@ -148,7 +148,7 @@ def _launch_demo(args, model, tokenizer):
|
|
148 |
|
149 |
if 'image' in query[0]:
|
150 |
image = tokenizer.draw_bbox_on_latest_picture(response)
|
151 |
-
|
152 |
image_path = save_image(image, uploaded_file_dir)
|
153 |
_chatbot[-1] = (chat_query, (image_path,))
|
154 |
else:
|
|
|
148 |
|
149 |
if 'image' in query[0]:
|
150 |
image = tokenizer.draw_bbox_on_latest_picture(response)
|
151 |
+
if image is not None:
|
152 |
image_path = save_image(image, uploaded_file_dir)
|
153 |
_chatbot[-1] = (chat_query, (image_path,))
|
154 |
else:
|