Harsh2001 commited on
Commit
e2de233
·
verified ·
1 Parent(s): 7e64201

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -47,5 +47,5 @@ def get_answer(query):
47
 
48
  # Decode the generated text
49
  generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
50
- chat_history.append(Document(page_content=generated_text.split(query)[1].capitalize()))
51
  return generated_text.split(query)[1].capitalize()
 
47
 
48
  # Decode the generated text
49
  generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
50
+ chat_history.append(Document(page_content=generated_text.split(query)[1].replace('.', '. ').capitalize()))
51
  return generated_text.split(query)[1].capitalize()