Update app.py

app.py

@@ -1,6 +1,6 @@
 import transformers
 import streamlit as st
-
+from nltk import sent_tokenize
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import json
 
@@ -33,6 +33,7 @@ def infer(input_ids, max_length, temperature, top_k, top_p):
 
 
 def chunking(book_text):
+    sentences = sent_tokenize(book_text)
     segments = [] 
     #sentences, token_lens
     current_segment = ""