Update utils.py
Browse files
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()
|