Update app.py

app.py

@@ -95,6 +95,9 @@ _book = test_book[book_index]['book'][:10000]
 st.title("Book Summarization 📚")
 st.write("Book Summarization using BART-BOOK! ")
 #book_index = st.sidebar.slider("Select Book Example", value = 0,min_value = 0, max_value=4)
+
+docu_size = len(tokenizer.tokenize(_book))
+st.write(f"Document size: {docu_size} tokens")
 sent = st.text_area("Text", _book, height = 1500)
 max_length = st.sidebar.slider("Max Length", value = 512,min_value = 10, max_value=1024)
 temperature = st.sidebar.slider("Temperature", value = 1.0, min_value = 0.0, max_value=1.0, step=0.05)
@@ -139,4 +142,7 @@ for i in range(10):
        
     summaries = generate_output(chunked_segments[i])
     st.write(f'A summary of Segment {i}.')
+    
+    chunk_size = len(tokenizer.tokenize(summaries))
+    st.write(f"Summary of Segment's size: {chunk_size} tokens")
     st.success(summaries[-1])