added diversity to key sentence extraction

.gitignore

@@ -0,0 +1 @@
+/__pycache__/

app.py

@@ -11,9 +11,11 @@ paraphraser = ParaphraseModel()
 #    ('T5-base', 'DistilT5-base', 'T5-small')
 #)
 
+top_n = st.sidebar.slider('Top_n', 1, 20, 5)
+diversity = st.sidebar.slider('Diversity', 0.0, 1.0, 0.6)
 
-top_k = st.sidebar.slider('Top_K', 100, 300, 168)
 
+top_k = st.sidebar.slider('Top_K', 100, 300, 168)
 top_p = st.sidebar.slider('Top_P', 0.0, 1.0, 0.95)
 
 st.header("Bullet-point Summarization")
@@ -22,7 +24,7 @@ st.header("Bullet-point Summarization")
 txt = st.text_area('Text to analyze', )
 
 if len(txt) >= 1:
-    key_sentences = get_key_sentences(txt)
+    key_sentences = get_key_sentences(txt, top_n=top_n, diversity=('mmr', diversity))
     sentences = []
     for i in sorted(key_sentences):
         sentences.append(key_sentences[i])