Update utils.py
Browse files
utils.py
CHANGED
@@ -47,4 +47,4 @@ def get_answer(query):
|
|
47 |
# Decode the generated text
|
48 |
generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
49 |
chat_history.append(Document(page_content=generated_text))
|
50 |
-
return generated_text.capitalize()
|
|
|
47 |
# Decode the generated text
|
48 |
generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
49 |
chat_history.append(Document(page_content=generated_text))
|
50 |
+
return generated_text.split(query)[1].capitalize()
|