WRX020510 commited on
Commit
6f04bba
·
verified ·
1 Parent(s): 3ee3a5a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,8 +12,8 @@ def main():
12
  # Load sentiment analysis pipeline only once using st.session_state
13
  if 'sentiment_analyzer' not in st.session_state:
14
  # Load your custom sentiment analysis model
15
- model = AutoModelForSequenceClassification.from_pretrained("path_to/CustomModel_twitter", num_labels=3)
16
- tokenizer = AutoTokenizer.from_pretrained("path_to/CustomModel_twitter")
17
  st.session_state.sentiment_analyzer = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
18
 
19
  # Load summarization pipeline only once using st.session_state
 
12
  # Load sentiment analysis pipeline only once using st.session_state
13
  if 'sentiment_analyzer' not in st.session_state:
14
  # Load your custom sentiment analysis model
15
+ model = AutoModelForSequenceClassification.from_pretrained("WRX020510/CustomModel_twitter", num_labels=3)
16
+ tokenizer = AutoTokenizer.from_pretrained("cardiffnlp/twitter-roberta-base-sentiment-latest")
17
  st.session_state.sentiment_analyzer = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
18
 
19
  # Load summarization pipeline only once using st.session_state