Update README.md
Browse files
README.md
CHANGED
|
@@ -37,6 +37,22 @@ Following languages supported by the model:
|
|
| 37 |
- spanish (es)
|
| 38 |
- french (fr)
|
| 39 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 40 |
## Model Training
|
| 41 |
|
| 42 |
Epoch Training Loss Validation Loss
|
|
|
|
| 37 |
- spanish (es)
|
| 38 |
- french (fr)
|
| 39 |
|
| 40 |
+
# Use a pipeline as a high-level helper
|
| 41 |
+
from transformers import pipeline
|
| 42 |
+
|
| 43 |
+
```python
|
| 44 |
+
pipe = pipeline("text-classification", model="ImranzamanML/GEFS-language-detector")
|
| 45 |
+
```
|
| 46 |
+
|
| 47 |
+
# Load model directly
|
| 48 |
+
|
| 49 |
+
```python
|
| 50 |
+
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
| 51 |
+
|
| 52 |
+
tokenizer = AutoTokenizer.from_pretrained("ImranzamanML/GEFS-language-detector")
|
| 53 |
+
model = AutoModelForSequenceClassification.from_pretrained("ImranzamanML/GEFS-language-detector")
|
| 54 |
+
```
|
| 55 |
+
|
| 56 |
## Model Training
|
| 57 |
|
| 58 |
Epoch Training Loss Validation Loss
|