Harsh2001 commited on
Commit
d306c41
·
verified ·
1 Parent(s): 031f4be

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
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()