From c8e5ec7e387e63ded0690364cfe71f28486be12b Mon Sep 17 00:00:00 2001
From: Zhijie Lin <136380243@qq.com>
Date: Thu, 25 May 2023 17:33:52 +0800
Subject: [PATCH] test

---
 demo.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/demo.py b/demo.py
index d5c3ef0..3fd4008 100644
--- a/demo.py
+++ b/demo.py
@@ -114,7 +114,7 @@ def gradio_answer(image, chatbot, chat_state, emb_list, num_beams, temperature):
     #                           max_length=2000)[0]
     llm_message = "I don't know"
     chatbot[-1][1] = llm_message
-    ground_img = grounding.prompt2mask(image, 'dog')
+    ground_img = grounding.prompt2mask(image, chatbot[-1][0])
     return gr.update(value=ground_img, interactive=False), \
            chatbot, chat_state, emb_list