update cap

app.py

@@ -241,7 +241,7 @@ def gradio_ask(user_message, chatbot, chat_state, radio):
     # if len(user_message) == 0:
     #     return gr.update(interactive=True, placeholder='Input should not be empty!'), chatbot, chat_state
 
-    chat.ask(user_message, chat_state, radio, model_name)
+    chat.ask(user_message, chat_state, radio)
     chatbot = chatbot + [[user_message, None]]
     return chatbot, chat_state
 
@@ -250,13 +250,13 @@ def generate_ans(user_message, chatbot, chat_state, img_list, radio, text, num_b
     # if len(user_message) == 0:
     #     return gr.update(interactive=True, placeholder='Input should not be empty!'), chatbot, chat_state
 
-    chat.ask(user_message, chat_state, radio, model_name)
+    chat.ask(user_message, chat_state, radio)
     chatbot = chatbot + [[user_message, None]]
     # return chatbot, chat_state
     image = None
     llm_message, image = \
         chat.answer(conv=chat_state, img_list=img_list, max_new_tokens=300, num_beams=1, temperature=temperature,
-                    max_length=2000, radio=radio, text_input=text, model_name=model_name)
+                    max_length=2000, radio=radio, text_input=text)
 
     chatbot[-1][1] = llm_message
     if chat_state[-1]["from"] == "gpt":
@@ -273,7 +273,7 @@ def gradio_answer(chatbot, chat_state, img_list, radio, text, num_beams, tempera
     image = None
     llm_message, image = \
         chat.answer(conv=chat_state, img_list=img_list, max_new_tokens=300, num_beams=1, temperature=temperature,
-                    max_length=2000, radio=radio, text_input=text, model_name=model_name)
+                    max_length=2000, radio=radio, text_input=text)
 
     chatbot[-1][1] = llm_message
     if chat_state[-1]["from"] == "gpt":