Update app.py

app.py

@@ -34,7 +34,7 @@ def preprocess_text(x):
   
 @st.cache(allow_output_mutation=True)
 def extractive_model():    
-  model = Summarizer()
+  model = Summarizer('distilbert-base-uncased')
   return model
   
 @st.cache(allow_output_mutation=True)
@@ -91,9 +91,9 @@ if summarize:
   with st.spinner(text="Loading Model and creating summary. This might take a few seconds depending on the length of your text..."):
     model =  model()
     #summarized_text = text_to_summarize if len(text_to_summarize) > 60 else ''.join(model(body, min_length=60))
-    min_ = min(100,len(text_to_summarize)/5)
-    max_ = min(600,len(text_to_summarize)/2)
-    summarized_text = ''.join(model(text_to_summarize, min_length=min_,max_length=max_)) if summary_type == "Extractive" else model(text_to_summarize, min_length=min_,max_length=max_)[0]['summary_text']
+    min_ = min(80,len(text_to_summarize)/5)
+    max_ = min(400,len(text_to_summarize)/2)
+    summarized_text = ''.join(model(text_to_summarize, min_length=min_,max_length=max_,ratio=0.2)) if summary_type == "Extractive" else model(text_to_summarize, min_length=min_,max_length=max_)[0]['summary_text']
     
   st.subheader("Original text")
   st.write(text_to_summarize)