lindsay-qu commited on
Commit
07eab47
·
1 Parent(s): 203c680

Update core/chatbot/retrieval_chatbot.py

Browse files
Files changed (1) hide show
  1. core/chatbot/retrieval_chatbot.py +2 -1
core/chatbot/retrieval_chatbot.py CHANGED
@@ -60,7 +60,8 @@ class RetrievalChatbot(BaseChatbot):
60
  print("Query: {message}".format(message=message))
61
  question = self.decomposer.refine(message,image_path)
62
  print(question)
63
- sub_questions = ast.literal_eval(question)
 
64
  print("Decomposed your query into subquestions: {sub_questions}".format(sub_questions=sub_questions))
65
  references = ""
66
  for sub_question in sub_questions:
 
60
  print("Query: {message}".format(message=message))
61
  question = self.decomposer.refine(message,image_path)
62
  print(question)
63
+ question = question.replace('"', "'").replace("', '", "','").lstrip("['").rstrip("']")
64
+ sub_questions = question.split("','")
65
  print("Decomposed your query into subquestions: {sub_questions}".format(sub_questions=sub_questions))
66
  references = ""
67
  for sub_question in sub_questions: