Jason Surya commited on
Commit
d1777a1
1 Parent(s): dfdd890

add max length

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import streamlit as st
2
  from transformers import pipeline
3
 
4
  def summarizeText(summarizer, txt):
5
- summarized_text = summarizer(txt)[0]["summary_text"]
6
  return summarized_text
7
 
8
  @st.cache_resource
 
2
  from transformers import pipeline
3
 
4
  def summarizeText(summarizer, txt):
5
+ summarized_text = summarizer(txt, max_length=100)[0]["summary_text"]
6
  return summarized_text
7
 
8
  @st.cache_resource