Spaces:
Runtime error
Runtime error
File size: 1,629 Bytes
91e858d |
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 |
# Author: Ricardo Lisboa Santos
# Creation date: 2024-01-10
import streamlit as st
import time
import AI.sentiment_analysis as ai
def run():
st.set_page_config(page_title="Sentiment Analysis", page_icon="📈")
# setSidebar()
# st.write('Sentiment Analysis')
st.markdown("# Sentiment Analysis")
input = st.text_input('Enter your prompt here.')
if st.button('Click me to run'):
progress_bar = st.sidebar.progress(0)
status_text = st.sidebar.empty()
loading_text='Loading Model'
with st.spinner(text=loading_text):
status_text.text("Getting Device")
device = ai.getDevice("cpu")
progress_bar.progress(30)
status_text.text("Loading Model")
model = ai.loadClassifier(device)
progress_bar.progress(60)
status_text.text("Classifying")
output = ai.classify(model, input)
progress_bar.progress(90)
ai.clearCache("cpu", model)
progress_bar.progress(100)
status_text.text("Done")
if output[0].get('label') == 'NEGATIVE':
st.error(' 😔 ' + output[0].get('label'))
elif output[0].get('label') == 'POSITIVE':
st.success(' 😃 ' + output[0].get('label'))
# def setSidebar():
# st.sidebar.header("Sentiment Analysis")
# st.sidebar.write(
# """This demo illustrates a combination of plotting and animation with
# Streamlit. We're generating a bunch of random numbers in a loop for around
# 5 seconds. Enjoy!"""
# )
if __name__ == '__main__':
run() |