File size: 479 Bytes
5900be6
 
 
 
 
 
 
8859c67
5900be6
 
 
1a8b5dc
 
 
5900be6
 
 
 
 
 
8859c67
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
import awesome_streamlit as ast
import streamlit as st
import requests
import time 
from transformers import pipeline
import os



sentiment_pipeline = pipeline(task = "text-classification", model = "gritli/bert-sentiment-analyses-imdb")

user_input = st.text_input(
        "Enter your text here:",
    )

if user_input:
    result = sentiment_pipeline(user_input)
    sentiment = result[0]["label"]

    sentiment = label_dict[sentiment]
    st.write(f"Detection: {sentiment}")