Update main.py

main.py

@@ -25,8 +25,8 @@ def load_models():
                                      tokenizer=AutoTokenizer.from_pretrained("usmiva/bert-deepfake-bg"))
 
     st.session_state.emotions = pipeline(task="text-classification",
-                                     model=BertForSequenceClassification.from_pretrained("TRACES/emotions", use_auth_token=os.environ['ACCESS_TOKEN2'],  num_labels=2),
-                                     tokenizer=AutoTokenizer.from_pretrained("TRACES/emotions", use_auth_token=os.environ['ACCESS_TOKEN2']))
+                                     model=BertForSequenceClassification.from_pretrained("TRACES/emotions", use_auth_token=os.environ['ACCESS_TOKEN'],  num_labels=2),
+                                     tokenizer=AutoTokenizer.from_pretrained("TRACES/emotions", use_auth_token=os.environ['ACCESS_TOKEN']))
 
 
 def load_content():
@@ -159,5 +159,5 @@ else:
     st.write(content['disclaimer_title'][st.session_state.lang])
     st.write(content['disclaimer'][st.session_state.lang])
     if st.button(content['disclaimer_agree_text'][st.session_state.lang]):
-        st.session_state.agree = True 
+        st.session_state.agree = True
         st.experimental_rerun()