Update README.md

README.md

@@ -110,10 +110,27 @@ If you use this model in your research or applications, please cite our paper:
 ```python
 from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
 
+# Load tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained("bilalzafar/FinAI-BERT-IslamicBanks")
 model = AutoModelForSequenceClassification.from_pretrained("bilalzafar/FinAI-BERT-IslamicBanks")
 
+# Create the pipeline
 classifier = pipeline("text-classification", model=model, tokenizer=tokenizer)
-result = classifier("Our Shariah-compliant bank has deployed AI-driven credit risk assessment tools.")
-print(result)
-# Output: [{'label': 'LABEL_1', 'score': ...}]  # LABEL_1 = AI, LABEL_0 = Non-AI
+
+# Label mapping
+label_map = {
+    "LABEL_0": "Non-AI",
+    "LABEL_1": "AI"
+}
+
+# Input text
+text = "Our Shariah-compliant bank has deployed AI-driven credit risk assessment tools."
+
+# Run classification
+result = classifier(text)[0]
+
+# Output
+label = label_map.get(result['label'], result['label'])
+score = result['score']
+print(f"Classification: {label} | Score: {score:.4f}")
+