chiennv
/

langid-mmbert-small

Model card Files Files and versions

chiennv commited on Mar 14

Commit

a1461a1

·

verified ·

1 Parent(s): 28ab986

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -56,7 +56,7 @@ def fast_detect_unknown(text: str) -> bool:
 import torch
 from transformers import pipeline
-model_id = "chiennv/langid-mmbert-small-8gpu"
 device = 0 if torch.cuda.is_available() else -1
 clf = pipeline(
     "text-classification",
@@ -80,7 +80,7 @@ else:
 import torch
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
-model_id = "chiennv/langid-mmbert-small-8gpu"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

 import torch
 from transformers import pipeline
+model_id = "chiennv/langid-mmbert-small"
 device = 0 if torch.cuda.is_available() else -1
 clf = pipeline(
     "text-classification",
 import torch
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
+model_id = "chiennv/langid-mmbert-small"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")