saidsef commited on
Commit
e50c5bd
·
verified ·
1 Parent(s): 11225f5

feat: update to text summarisation

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,9 +1,9 @@
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
- pipe = pipeline(task="sentiment-analysis", model="distilbert/distilbert-base-uncased-finetuned-sst-2-english")
5
- text = st.text_area("what is the sentiment score?")
6
 
7
  if text:
8
  out = pipe(text)
9
- st.json(out)
 
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
+ pipe = pipeline(task="summarization", model="facebook/bart-large-cnn")
5
+ text = st.text_area("Summarise the following")
6
 
7
  if text:
8
  out = pipe(text)
9
+ st.write(out)