Update README.md

README.md

@@ -8,6 +8,12 @@ tags:
 - delivery
 - bert
 - bert-persian
+language:
+- fa
+metrics:
+- accuracy
+- f1
+pipeline_tag: text-classification
 ---
 # Persian Sentiment Analysis for Delivery Complaints
 
@@ -15,7 +21,7 @@ tags:
 This model is fine-tuned **HooshvareLab/bert-fa-base-uncased** model to classify Persian comments related to delivery complaints. The model predicts whether a comment is about a negative delivery experience or not.
 
 ## 📊 Dataset
-The model was fine-tuned using the **Labeled Persian Comments** dataset from Kaggle:
+The model was fine-tuned using the ** Basalam comments ** dataset from Kaggle:
 [📂 Basalam Comments Dataset](https://www.kaggle.com/datasets/alirezaazizkhani/labeled-persian-comments)
 
 This dataset contains Persian comments labeled as:
@@ -38,7 +44,7 @@ from transformers import AutoModelForSequenceClassification, AutoTokenizer
 import torch
 
 def classify_comment(text):
-    model_name = "alireza-2003/bert_fa_for_bad_delivery_detection"
+    model_name = "alireza-2003/bert_fa_bad_delivery"
     model = AutoModelForSequenceClassification.from_pretrained(model_name)
     tokenizer = AutoTokenizer.from_pretrained(model_name)