Update pages/21_NLP_Transformer.py

pages/21_NLP_Transformer.py

@@ -48,7 +48,7 @@ if st.sidebar.button("Train"):
     for epoch in range(num_epochs):
         for batch in train_dataloader:
             batch = {k: v.to(device) for k, v in batch.items()}
-            outputs = model(**batch, labels=batch["label"])  # Fixed input key
+            outputs = model(**batch, labels=batch["labels"])  # Fixed input key
             loss = outputs.loss
             loss.backward()