Update README.md
Browse files
README.md
CHANGED
|
@@ -64,6 +64,18 @@ The following hyperparameters were used during training:
|
|
| 64 |
- Datasets 2.17.0
|
| 65 |
- Tokenizers 0.15.1
|
| 66 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 67 |
# Contact
|
| 68 |
|
| 69 |
-
Please reach out to [luca.flammia@gmail.com] if you have any questions or feedback.
|
|
|
|
| 64 |
- Datasets 2.17.0
|
| 65 |
- Tokenizers 0.15.1
|
| 66 |
|
| 67 |
+
```python
|
| 68 |
+
from transformers import pipeline
|
| 69 |
+
|
| 70 |
+
classifier = pipeline("text-classification", model="LukeGPT88/imdb_text_classifier")
|
| 71 |
+
classifier("I see it and it was awesome.")
|
| 72 |
+
```
|
| 73 |
+
|
| 74 |
+
```python
|
| 75 |
+
Output:
|
| 76 |
+
[{'label': 'POSITIVE', 'score': 0.9958052635192871}]
|
| 77 |
+
```
|
| 78 |
+
|
| 79 |
# Contact
|
| 80 |
|
| 81 |
+
Please reach out to [luca.flammia@gmail.com](luca.flammia@gmail.com) if you have any questions or feedback.
|