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.
|