Spaces:
Sleeping
Sleeping
Update main.py
Browse files
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['
|
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 |
|