Update main.py

main.py

@@ -25,10 +25,11 @@ 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),
+                                     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']))
 
 
+
 def load_content():
     with open('resource/page_content.json', encoding='utf8') as json_file:
         return json.load(json_file)