File size: 3,056 Bytes
32c9084 a906ccd 32c9084 a906ccd 0e5f8d9 32c9084 a906ccd 9b7381f 32c9084 c9d9479 a906ccd b4e2d85 c9d9479 a906ccd 1cbfec0 2c0ab44 a906ccd dd63333 a906ccd f7d4880 a906ccd b4e2d85 a906ccd b4e2d85 f35d308 4426139 f35d308 4426139 f35d308 85c2bc8 f35d308 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 |
#x = st.slider('Select a value')
#st.write(x, 'squared is', x * x)
import streamlit as st
from transformers import pipeline
# Load the summarization model
#summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6") # smaller version of the model
summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
# Default article text
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.
"""
# Create a text area for user input
ARTICLE = st.sidebar.text_area('Enter Article', DEFAULT_ARTICLE, height=150)
# Define the summarization function
def summarize(txt):
st.write('\n\n')
st.write(txt[:100]) # Display the first 100 characters of the article
st.write('--------------------------------------------------------------')
summary = summarizer(txt, max_length=130, min_length=30, do_sample=False)
st.write(summary[0]['summary_text'])
# Create a button and trigger the summarize function when clicked
if st.sidebar.button('Summarize Article'):
summarize(ARTICLE)
else:
st.warning('👈 Please enter Article!')
#################################
# Initialize the sentiment analysis pipeline
# No model was supplied, defaulted to distilbert-base-uncased-finetuned-sst-2-english
sentiment_pipeline = pipeline("sentiment-analysis")
# Default article text
DEFAULT_SENTIMENT = [
"I'm so happy today!",
"This is the worst experience ever.",
"It's a decent product, nothing special."
]
# Create a text area for user input
SENTIMENT = st.sidebar.text_area('Enter Sentiment', DEFAULT_SENTIMENT, height=150)
# Define the summarization function
def summarize(txt):
st.write('\n\n')
#st.write(txt[:100]) # Display the first 100 characters of the article
st.write('--------------------------------------------------------------')
# Perform Hugging sentiment analysis on multiple texts
results = sentiment_pipeline(txt)
# Display the results
for i, text in enumerate(txt):
st.write(f"Text: {text}")
st.write(f"Sentiment: {results[i]['label']}, Score: {results[i]['score']:.2f}\n")
# Create a button and trigger the summarize function when clicked
if st.sidebar.button('Summarize Sentiment'):
summarize(ast.literal_eval(SENTIMENT)) #convert string to actual list
else:
st.warning('👈 Please enter Sentiment!') |