Spaces:
Runtime error
Runtime error
| # 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() |