saicharan1234 commited on
Commit
844350c
·
verified ·
1 Parent(s): 97d9864

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -114,7 +114,7 @@ async def process_item(
114
  llm_message = chat.answer(conv=chat_state, img_list=img_list, max_new_tokens=300, num_beams=1, temperature=1,
115
  max_length=2000)[0]
116
  chatbot[-1][1] = llm_message
117
- return chatbot, chat_state, img_list
118
 
119
 
120
  if __name__ == "__main__":
 
114
  llm_message = chat.answer(conv=chat_state, img_list=img_list, max_new_tokens=300, num_beams=1, temperature=1,
115
  max_length=2000)[0]
116
  chatbot[-1][1] = llm_message
117
+ return chatbot
118
 
119
 
120
  if __name__ == "__main__":