Update README.md
Browse files
README.md
CHANGED
@@ -42,6 +42,7 @@ converted to hinglish text.
|
|
42 |
|
43 |
The model has a test loss of 0.6 and an f1 score of 0.74 on the unseen data from the dataset.
|
44 |
|
|
|
45 |
```
|
46 |
# Use a pipeline as a high-level helper
|
47 |
from transformers import pipeline
|
@@ -49,10 +50,9 @@ from transformers import pipeline
|
|
49 |
pipe = pipeline("text-classification", model="pascalrai/hinglish-twitter-roberta-base-sentiment")
|
50 |
pipe("tu mujhe pasandh heh")
|
51 |
```
|
52 |
-
|
53 |
```
|
54 |
# Load model directly
|
55 |
-
# Load model directly
|
56 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
57 |
import torch
|
58 |
|
|
|
42 |
|
43 |
The model has a test loss of 0.6 and an f1 score of 0.74 on the unseen data from the dataset.
|
44 |
|
45 |
+
## Model Inference using pipeline
|
46 |
```
|
47 |
# Use a pipeline as a high-level helper
|
48 |
from transformers import pipeline
|
|
|
50 |
pipe = pipeline("text-classification", model="pascalrai/hinglish-twitter-roberta-base-sentiment")
|
51 |
pipe("tu mujhe pasandh heh")
|
52 |
```
|
53 |
+
## Model Inference
|
54 |
```
|
55 |
# Load model directly
|
|
|
56 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
57 |
import torch
|
58 |
|