Fixed a bug with the output function return_top_k

pages/2_Context-based_chatbot.py

@@ -42,7 +42,7 @@ def return_top_k(sentence, word=None, rels=False):
 #     inputs = [f"Description : {sentence} Word : "]
     
   output = model(sentence) 
-  output = [output[i]['token_str'] for i in output.keys()]
+  output = [output[i]['token_str'].strip() for i in range(len(output))]
   return output