Update README.md

README.md

@@ -91,7 +91,10 @@ import torch
 from transformers import pipeline
 device = torch.device("mps" if torch.backends.mps.is_available() else ("cuda" if torch.cuda.is_available() else "cpu"))
 
-pipe = pipeline("text-classification", model="GEODE/bert-base-multilingual-cased-geography-entry-classification", device=device)
+tokenizer = AutoTokenizer.from_pretrained("GEODE/bert-base-multilingual-cased-geography-entry-classification")
+model = AutoModelForSequenceClassification.from_pretrained("GEODE/bert-base-multilingual-cased-geography-entry-classification")
+
+pipe = pipeline("text-classification", model=model, tokenizer=tokenizer,  truncation=True, device=device)
 
 samples = [
     "* ALBI, (Géog.) ville de France, capitale de l'Albigeois, dans le haut Languedoc : elle est sur le Tarn. Long. 19. 49. lat. 43. 55. 44.",