Update README.md

README.md

@@ -6,6 +6,8 @@ metrics:
 - accuracy
 library_name: transformers
 pipeline_tag: text-classification
+datasets:
+- GautamR/grievance_agri
 ---
 Model to classify grievances into 3 buckets : 
 The  buckets are : ['Account number is not Correct' ,'Installment not received', 'Others']
@@ -28,6 +30,4 @@ Epoch 5/15 - Loss: 0.1612 - Accuracy: 1.0000 - F1 Score: 1.0000
 Epoch 6/15 - Loss: 0.0883 - Accuracy: 1.0000 - F1 Score: 1.0000
 Epoch 7/15 - Loss: 0.0592 - Accuracy: 1.0000 - F1 Score: 1.0000
 Epoch 8/15 - Loss: 0.0418 - Accuracy: 1.0000 - F1 Score: 1.0000
-Epoch 9/15 - Loss: 0.0321 - Accuracy: 1.0000 - F1 Score: 1.0000
-
-
+Epoch 9/15 - Loss: 0.0321 - Accuracy: 1.0000 - F1 Score: 1.0000