Update app.py

app.py

@@ -1,6 +1,7 @@
 from transformers import pipeline
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import gradio as gr
+from nltk.tokenize import sent_tokenize
 
 model = "janny127/autotrain-7qmts-cs1er"
 tokenizer = AutoTokenizer.from_pretrained(model)
@@ -31,9 +32,12 @@ def generate_answer(query, sample_num=3):
  for seq in sequences:
  answer = seq['generated_text'].replace(formatted_prompt, "")
  answers.append(answer)
- #print(f"Result: {answer}")
- #print("------------------------------------------")
- return answers
+ answer = sent_tokenize(answers[0])
+ final_answer = ''
+ for an in answer:
+ final_answer += an.strip()
+ return final_answer
+ 
 
 interface = gr.ChatInterface(
  fn=generate_answer,