Update app.py

app.py

@@ -48,25 +48,29 @@ class ChatBot:
         else:
             return None
 
+    def clean_response(self, response):
+        return re.sub(r'<ref>(.*?)</ref>(?:<box>.*?</box>)*(?:<quad>.*?</quad>)*', r'\1', response).strip()
+
     def clear_memory(self):
         if torch.cuda.is_available():
             torch.cuda.empty_cache()
         gc.collect()
 
-def chat_interface(chatbot, text_query, file):
+def chat_interface(text_query, file):
+    chatbot = ChatBot()
     image_path = file.name if file is not None else None
     response = chatbot.chat(image_path=image_path, text_query=text_query)
+
     if "<box>" in response:
         image_with_boxes = chatbot.draw_boxes(response)
+        text_response = chatbot.clean_response(response)
         chatbot.clear_memory()
-        text_response = re.sub(r'<ref>.*?</ref>(?:<box>.*?</box>)*(?:<quad>.*?</quad>)*', '', response).strip()
         return [("Qwen-VL_Chat", text_response), ("Qwen-VL_Image", image_with_boxes)]
     else:
         chatbot.clear_memory()
         return [("Qwen-VL_Chat", response)]
-
+        
 with gr.Blocks() as demo:
-    chatbot = ChatBot()
     gr.Markdown("""
 # 🙋🏻‍♂️欢迎来到🌟Tonic 的🦆Qwen-VL-Chat🤩Bot！🚀
 # 🙋🏻‍♂️Welcome to Tonic's🦆Qwen-VL-Chat🤩Bot！🚀
@@ -87,7 +91,7 @@ Join us:  TeamTonic  is always making cool demos! Join our active builder's comm
     
     submit_btn.click(
         fn=chat_interface,
-        inputs=[chatbot, query, file_upload],
+        inputs=[query, file_upload],
         outputs=[chatbot_component]
     )
     gr.Markdown("""