Update app.py

app.py

@@ -45,6 +45,13 @@ def initialize_model_and_tokenizer(model_name, num_labels):
     try:
         tokenizer = AutoTokenizer.from_pretrained(model_name)
         model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=num_labels)
+        
+        # Set the padding token to the EOS token
+        tokenizer.pad_token = tokenizer.eos_token
+        
+        # Resize the token embeddings as we added a new token
+        model.resize_token_embeddings(len(tokenizer))
+        
         return tokenizer, model
     except Exception as e:
         st.error(f"Error initializing model and tokenizer: {str(e)}")