import streamlit as st import transformers from transformers import pipeline st.title("Text Classification using BERT") text = st.text_input("Enter your text") classifier = pipeline('text-classification', model='bert-base-uncased', tokenizer='bert-base-uncased') label_map = {0: 'Negative', 1: 'Positive'} label_id = result[0]['label'] label_name = label_map[label_id] if st.button("Classify"): result = classifier(text) st.write("Classification Label:", label_name)