|
|
|
|
|
|
|
import streamlit as st |
|
from transformers import pipeline |
|
import ast |
|
|
|
|
|
|
|
summarizer = pipeline("summarization", model="facebook/bart-large-cnn") |
|
|
|
|
|
DEFAULT_ARTICLE = """ New York (CNN)When Liana Barrientos was 23 years old, she got married in Westchester County, New York. |
|
A year later, she got married again in Westchester County, but to a different man and without divorcing her first husband. |
|
Only 18 days after that marriage, she got hitched yet again. Then, Barrientos declared "I do" five more times, sometimes only within two weeks of each other. |
|
In 2010, she married once more, this time in the Bronx. In an application for a marriage license, she stated it was her "first and only" marriage. |
|
Barrientos, now 39, is facing two criminal counts of "offering a false instrument for filing in the first degree," referring to her false statements on the |
|
2010 marriage license application, according to court documents. |
|
""" |
|
|
|
|
|
ARTICLE = st.sidebar.text_area('Enter Article', DEFAULT_ARTICLE, height=150) |
|
|
|
|
|
def summarize(txt): |
|
st.write('\n\n') |
|
st.write(txt[:100]) |
|
st.write('--------------------------------------------------------------') |
|
summary = summarizer(txt, max_length=130, min_length=30, do_sample=False) |
|
st.write(summary[0]['summary_text']) |
|
|
|
|
|
if st.sidebar.button('Summarize Article'): |
|
summarize(ARTICLE) |
|
else: |
|
st.warning('π Please enter Article!') |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
sentiment_pipeline = pipeline("sentiment-analysis") |
|
|
|
|
|
DEFAULT_SENTIMENT = [ |
|
"I'm so happy today!", |
|
"This is the worst experience ever.", |
|
"It's a decent product, nothing special." |
|
] |
|
|
|
|
|
SENTIMENT = st.sidebar.text_area('Enter Sentiment', DEFAULT_SENTIMENT, height=150) |
|
|
|
|
|
def summarize(txt): |
|
st.write('\n\n') |
|
|
|
st.write('--------------------------------------------------------------') |
|
|
|
|
|
results = sentiment_pipeline(txt) |
|
|
|
|
|
for i, text in enumerate(txt): |
|
st.write(f"Text: {text}") |
|
st.write(f"Sentiment: {results[i]['label']}, Score: {results[i]['score']:.2f}\n") |
|
|
|
|
|
|
|
if st.sidebar.button('Summarize Sentiment'): |
|
|
|
|
|
|
|
summarize(ast.literal_eval(SENTIMENT)) |
|
else: |
|
st.warning('π Please enter Sentiment!') |