Daniel Tse commited on
Commit
f8474ce
·
1 Parent(s): f3e24b8

Change to use Facebook Summarizer

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,9 +4,9 @@ from transformers import pipeline
4
  x = st.slider('Select a value')
5
  st.write(x, 'squared is', x * x)
6
 
7
- pipe = pipeline('sentiment-analysis')
8
  text = st.text_area('enter some text')
9
 
10
  if text:
11
- out = pipe(text)
12
  st.json(out)
 
4
  x = st.slider('Select a value')
5
  st.write(x, 'squared is', x * x)
6
 
7
+ pipe = pipeline("summarization", model='facebook/bart-large-cnn')
8
  text = st.text_area('enter some text')
9
 
10
  if text:
11
+ out = pipe(text, max_length=130, min_length=30, do_sample=False)
12
  st.json(out)