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, ignore_mismatched_sizes=True),
-                                     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_TOKEN2'],  num_labels=2),
+                                     tokenizer=AutoTokenizer.from_pretrained("usmiva/bert-web-bg"))