Update README.md
Browse files
README.md
CHANGED
|
@@ -45,7 +45,7 @@ It classifies reviews as **Positive** ⭐ or **Negative** 👎 with high accurac
|
|
| 45 |
```python
|
| 46 |
from transformers import pipeline
|
| 47 |
|
| 48 |
-
classifier = pipeline("text-classification", model="
|
| 49 |
|
| 50 |
print(classifier("The burger was amazing, I'll definitely come back!"))
|
| 51 |
# [{'label': 'Positive', 'score': 0.998}]
|
|
|
|
| 45 |
```python
|
| 46 |
from transformers import pipeline
|
| 47 |
|
| 48 |
+
classifier = pipeline("text-classification", model="itserphan/yelp-polarity-roberta")
|
| 49 |
|
| 50 |
print(classifier("The burger was amazing, I'll definitely come back!"))
|
| 51 |
# [{'label': 'Positive', 'score': 0.998}]
|