Update README.md

README.md

@@ -123,9 +123,10 @@ from transformers import pipeline
 
 classifier = pipeline("text-classification", model="DiligentAI/urlbert-url-classifier")
 result = classifier("https://store.myshopify.com")
-print(result)
-# [{'label': 'LABEL_1', 'score': 0.7596}]
+
+
 Pydantic Integration (Production-Ready)
+
 from transformers import pipeline
 from pydantic import BaseModel, Field
 from typing import Literal
@@ -146,7 +147,6 @@ def classify_url(url: str) -> URLClassificationResult:
         label=label_map[result["label"]],
         confidence=result["score"]
     )
-```
 
 Limitations and Bias
 Max URL Length: Model trained on 64-token sequences. Longer URLs are truncated.
@@ -171,6 +171,7 @@ Language detection
 Spam filtering
 Model Card Authors
 DiligentAI Team
+
 Citation
 @misc{urlbert-classifier-2025,
   author = {DiligentAI},
@@ -198,6 +199,7 @@ Model checkpoint (.safetensors)
 Tokenizer configuration
 Label mapping (label_map.json)
 Performance metrics (metrics.json)
+
 Contact
 For issues, questions, or feedback:
 GitHub: DiligentAI/url-classifier