Update README.md

README.md

@@ -122,15 +122,12 @@ The following hyperparameters were used during training:
 ## How to Use
 
 ```python
+import torch
 from transformers import AutoTokenizer, AutoModelForTokenClassification
 from transformers import pipeline
 
-# Load model and tokenizer
-tokenizer = AutoTokenizer.from_pretrained("MatteoFasulo/ModernBERT-base-NER")
-model = AutoModelForTokenClassification.from_pretrained("MatteoFasulo/ModernBERT-base-NER")
-
 # Create NER pipeline
-ner_pipeline = pipeline("ner", model=model, tokenizer=tokenizer, aggregation_strategy="simple")
+ner_pipeline = pipeline("token-classification", model="MatteoFasulo/ModernBERT-base-NER", aggregation_strategy="simple", dtype=torch.bfloat16)
 
 # Example usage
 text = "Apple Inc. was founded by Steve Jobs in Cupertino, California."