Update README.md
Browse files
README.md
CHANGED
@@ -49,6 +49,8 @@ from transformers import pipeline
|
|
49 |
|
50 |
pipe = pipeline("text-classification", model="pascalrai/hinglish-twitter-roberta-base-sentiment")
|
51 |
pipe("tu mujhe pasandh heh")
|
|
|
|
|
52 |
```
|
53 |
## Model Inference
|
54 |
```
|
@@ -66,4 +68,14 @@ p = torch.nn.Softmax(dim = 1)(outputs.logits)
|
|
66 |
for index, each in enumerate(p.detach().numpy()):
|
67 |
print(f"Text: {inputs[index]}")
|
68 |
print(f"Negative: {round(float(each[0]),2)}\nNeutral: {round(float(each[1]),2)}\nPositive: {round(float(each[2]),2)}\n")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
69 |
```
|
|
|
49 |
|
50 |
pipe = pipeline("text-classification", model="pascalrai/hinglish-twitter-roberta-base-sentiment")
|
51 |
pipe("tu mujhe pasandh heh")
|
52 |
+
|
53 |
+
[{'label': 'positive', 'score': 0.7615439891815186}]
|
54 |
```
|
55 |
## Model Inference
|
56 |
```
|
|
|
68 |
for index, each in enumerate(p.detach().numpy()):
|
69 |
print(f"Text: {inputs[index]}")
|
70 |
print(f"Negative: {round(float(each[0]),2)}\nNeutral: {round(float(each[1]),2)}\nPositive: {round(float(each[2]),2)}\n")
|
71 |
+
|
72 |
+
Text: tum kon ho bhai
|
73 |
+
Negative: 0.02
|
74 |
+
Neutral: 0.91
|
75 |
+
Positive: 0.07
|
76 |
+
|
77 |
+
Text: tu mujhe pasandh heh
|
78 |
+
Negative: 0.01
|
79 |
+
Neutral: 0.22
|
80 |
+
Positive: 0.76
|
81 |
```
|