Update README.md
Browse files
README.md
CHANGED
|
@@ -41,6 +41,20 @@ More information needed
|
|
| 41 |
|
| 42 |
More information needed
|
| 43 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 44 |
## Training and evaluation data
|
| 45 |
|
| 46 |
More information needed
|
|
|
|
| 41 |
|
| 42 |
More information needed
|
| 43 |
|
| 44 |
+
## How to Use
|
| 45 |
+
|
| 46 |
+
```Python
|
| 47 |
+
from transformers import pipeline
|
| 48 |
+
|
| 49 |
+
model_name = "avanishd/distilbert-base-uncased-finetuned-imdb"
|
| 50 |
+
|
| 51 |
+
classifier = pipeline("text-classification", model=model_name)
|
| 52 |
+
|
| 53 |
+
text = "I am happy"
|
| 54 |
+
|
| 55 |
+
prediction = classifier(text)
|
| 56 |
+
```
|
| 57 |
+
|
| 58 |
## Training and evaluation data
|
| 59 |
|
| 60 |
More information needed
|