NursNurs commited on
Commit
a190b4a
·
1 Parent(s): 29fb3ce

fixed printing results bug

Browse files
Files changed (1) hide show
  1. pages/2_Context-based_chatbot.py +1 -2
pages/2_Context-based_chatbot.py CHANGED
@@ -230,8 +230,7 @@ if prompt_context:
230
  elif (st.session_state.messages_context[-2]['content'] == "Please give a sentence using a <mask> instead of the word you have in mind!") & (st.session_state.messages_context[-1]['content'] != "no"):
231
  write_bot("Great! Let me think what it could be...")
232
  st.session_state.descriptions_context.append(prompt_context)
233
- st.session_state.results_context['results_context'] = return_top_k_context
234
- (st.session_state.descriptions_context[-1])
235
  st.session_state.results_context['results_context_print'] = dict(zip(range(1, len(st.session_state.results_context['results_context'])+1), st.session_state.results_context['results_context']))
236
  write_bot("I think I have some ideas. Do you want to see my guesses or do you want a cue?")
237
  st.session_state.actions_context.append("result")
 
230
  elif (st.session_state.messages_context[-2]['content'] == "Please give a sentence using a <mask> instead of the word you have in mind!") & (st.session_state.messages_context[-1]['content'] != "no"):
231
  write_bot("Great! Let me think what it could be...")
232
  st.session_state.descriptions_context.append(prompt_context)
233
+ st.session_state.results_context['results_context'] = return_top_k_context(st.session_state.descriptions_context[-1])
 
234
  st.session_state.results_context['results_context_print'] = dict(zip(range(1, len(st.session_state.results_context['results_context'])+1), st.session_state.results_context['results_context']))
235
  write_bot("I think I have some ideas. Do you want to see my guesses or do you want a cue?")
236
  st.session_state.actions_context.append("result")