Update app.py

app.py

@@ -31,7 +31,7 @@ def infer(input_ids, max_length, temperature, top_k, top_p):
         top_p=top_p,
         do_sample=True,
         num_return_sequences=1,
-        num_beams=4,
+        #num_beams=4,
         no_repeat_ngram_size=2
     )
     
@@ -89,13 +89,13 @@ def chunking(book_text):
 
 
 book_index = 0
-_book = test_book[book_index]['book']
+_book = test_book[book_index]['book'][:10000]
 
 #prompts
 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)
-sent = st.text_area("Text", _book, height = 3000)
+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)
 top_k = st.sidebar.slider("Top-k", min_value = 0, max_value=5, value = 0)