varun500 commited on
Commit
2020d9c
1 Parent(s): c04970e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -12,9 +12,12 @@ def main():
12
  # Get user input
13
  doc = st.text_area("Document")
14
 
 
 
 
15
  # Extract keywords
16
  if st.button("Extract Keywords"):
17
- keywords = kw_model.extract_keywords(doc)
18
 
19
  # Get user choice for MMR
20
  apply_mmr = st.checkbox("Apply Maximal Marginal Relevance (MMR)")
 
12
  # Get user input
13
  doc = st.text_area("Document")
14
 
15
+ # Get user choice for stopwords removal
16
+ remove_stopwords = st.checkbox("Remove Stopwords")
17
+
18
  # Extract keywords
19
  if st.button("Extract Keywords"):
20
+ keywords = kw_model.extract_keywords(doc, stop_words=None if remove_stopwords else "english")
21
 
22
  # Get user choice for MMR
23
  apply_mmr = st.checkbox("Apply Maximal Marginal Relevance (MMR)")