lmoncla commited on
Commit
30fc74f
·
verified ·
1 Parent(s): c88c2b4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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, 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.",
 
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.",