640510702phithak's picture
Update app.py
1a21231
Raw History Blame Contribute Delete
580 Bytes
import torch
import streamlit as st
from transformers import pipeline
title="Sentiment Analysis"
classifier = pipeline(task="sentiment-analysis",
model="distilbert-base-uncased-finetuned-sst-2-english")
def main():
st.title(title)
with st.form("questionaire"):
prompt1 = st.text_input("your text")
clicked = st.form_submit_button("start")
if clicked:
results=classifier(prompt1)
st.text(results[0]['label'])
st.text(f"Score : {results[0]['score']}")
if __name__=='__main__':
main()