Update app.py

app.py

@@ -59,6 +59,9 @@ def initialize_model_and_tokenizer(model_name, num_labels):
         # Set the padding token to the EOS token
         tokenizer.pad_token = tokenizer.eos_token
         
+        # Update the model config
+        model.config.pad_token_id = tokenizer.pad_token_id
+        
         # Resize the token embeddings as we added a new token
         model.resize_token_embeddings(len(tokenizer))