Skip to content

Commit

Permalink
Make transformers install check positive (huggingface#7473)
Browse files Browse the repository at this point in the history
When transformers is correctly installed, you should get a positive message ^_^
  • Loading branch information
FremyCompany authored Sep 30, 2020
1 parent 7a0cf0e commit cc4eff8
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions docs/source/installation.md
Original file line number Diff line number Diff line change
Expand Up @@ -37,13 +37,13 @@ pip install transformers[tf-cpu]
To check 🤗 Transformers is properly installed, run the following command:

```bash
python -c "from transformers import pipeline; print(pipeline('sentiment-analysis')('I hate you'))"
python -c "from transformers import pipeline; print(pipeline('sentiment-analysis')('we love you'))"
```

It should download a pretrained model then print something like

```bash
[{'label': 'NEGATIVE', 'score': 0.9991129040718079}]
[{'label': 'POSITIVE', 'score': 0.9998704791069031}]
```

(Note that TensorFlow will print additional stuff before that last statement.)
Expand Down

0 comments on commit cc4eff8

Please sign in to comment.