Update README.md
Browse files
README.md
CHANGED
|
@@ -126,7 +126,7 @@ device = torch.device("mps" if torch.backends.mps.is_available() else ("cuda" if
|
|
| 126 |
tokenizer = AutoTokenizer.from_pretrained("GEODE/bert-base-multilingual-cased-place-entry-classification")
|
| 127 |
model = AutoModelForSequenceClassification.from_pretrained("GEODE/bert-base-multilingual-cased-place-entry-classification")
|
| 128 |
|
| 129 |
-
pipe = pipeline("text-classification", model=model, tokenizer=tokenizer,
|
| 130 |
|
| 131 |
samples = [
|
| 132 |
"* 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.",
|
|
|
|
| 126 |
tokenizer = AutoTokenizer.from_pretrained("GEODE/bert-base-multilingual-cased-place-entry-classification")
|
| 127 |
model = AutoModelForSequenceClassification.from_pretrained("GEODE/bert-base-multilingual-cased-place-entry-classification")
|
| 128 |
|
| 129 |
+
pipe = pipeline("text-classification", model=model, tokenizer=tokenizer, truncation=True, device=device)
|
| 130 |
|
| 131 |
samples = [
|
| 132 |
"* 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.",
|