Spaces:

Olivernyu
/

sentiment_analysis_app

Runtime error

Oliver Li commited on Apr 24, 2023

Commit

5dd2487

1 Parent(s): e7c1530

bug fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,9 +1,9 @@
 import streamlit as st
 import pandas as pd
 from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
-import SessionState
 # Function to load the pre-trained model
-ss = SessionState.get(table_df=pd.DataFrame(), first_run = True)
 def load_finetune_model(model_name):
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     model = AutoModelForSequenceClassification.from_pretrained(model_name)

 import streamlit as st
 import pandas as pd
 from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
 # Function to load the pre-trained model
+ss = st.session_state.get(table_df=pd.DataFrame(), first_run = True)
 def load_finetune_model(model_name):
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     model = AutoModelForSequenceClassification.from_pretrained(model_name)