TRACES commited on
Commit
1053a3c
1 Parent(s): 9c598cb

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -25,7 +25,7 @@ def load_models():
25
  tokenizer=AutoTokenizer.from_pretrained("usmiva/bert-deepfake-bg"))
26
 
27
  st.session_state.emotions = pipeline(task="text-classification",
28
- model=BertForSequenceClassification.from_pretrained("TRACES/emotions", use_auth_token=os.environ['ACCESS_TOKEN2'], num_labels=11),
29
  tokenizer=AutoTokenizer.from_pretrained("usmiva/bert-web-bg"))
30
 
31
 
 
25
  tokenizer=AutoTokenizer.from_pretrained("usmiva/bert-deepfake-bg"))
26
 
27
  st.session_state.emotions = pipeline(task="text-classification",
28
+ model=BertForSequenceClassification.from_pretrained("TRACES/emotions", use_auth_token=os.environ['ACCESS_TOKEN3'], num_labels=11),
29
  tokenizer=AutoTokenizer.from_pretrained("usmiva/bert-web-bg"))
30
 
31