| |
| |
|
|
| 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." |
| ]""" |
|
|
| DEFAULT_SENTIMENT = "I'm so happy today!" |
| |
| SENTIMENT = st.sidebar.text_area('Enter Sentiment', DEFAULT_SENTIMENT, height=150) |
|
|
| def is_valid_list_string(string): |
| try: |
| result = ast.literal_eval(string) |
| return isinstance(result, list) |
| except (ValueError, SyntaxError): |
| return False |
| |
| |
| def summarize(txt): |
| |
| st.write('\n\n') |
| |
| st.write('--------------------------------------------------------------') |
| |
| |
| |
| |
| if is_valid_list_string(txt): |
| txt_converted = ast.literal_eval(txt) |
| |
| results = sentiment_pipeline(txt_converted) |
| for i, text in enumerate(txt_converted): |
| st.write(f"Text: {text}") |
| st.write(f"Sentiment: {results[i]['label']}, Score: {results[i]['score']:.2f}\n") |
| else: |
| |
| results = sentiment_pipeline(txt) |
| st.write(f"Text: {txt}") |
| st.write(f"Sentiment: {results[0]['label']}, Score: {results[0]['score']:.2f}\n") |
|
|
| |
| if st.sidebar.button('Summarize Sentiment'): |
| |
| |
| |
| |
| summarize(SENTIMENT) |
| else: |
| st.warning('π Please enter Sentiment!') |