| import streamlit as st | |
| from transformers import pipeline | |
| # Create a pipeline for text classification | |
| pipe = pipeline("text-classification", model="distilbert/distilbert-base-uncased-finetuned-sst-2-english") | |
| # Streamlit app structure | |
| st.title("Text Classification App") | |
| st.write("Enter some text and the model will predict its sentiment:") | |
| # Input text from the user | |
| user_input = st.text_input("Input Text:") | |
| # When the button is clicked, classify the input | |
| if st.button("Classify"): | |
| if user_input: | |
| result = pipe(user_input) | |
| st.write(result) | |
| else: | |
| st.write("Please enter some text.") | |