traberph commited on
Commit
4f60dd3
1 Parent(s): 8b678e0

Update README.md

Browse files

added transformer requirement to model card

Files changed (1) hide show
  1. README.md +5 -1
README.md CHANGED
@@ -11,6 +11,10 @@ pipeline_tag: text-classification
11
  ## Reddit post classification
12
 
13
  This model predicts the subreddit of a provided post
 
 
 
 
14
 
15
  ```py
16
  from transformers import pipeline
@@ -20,7 +24,7 @@ pipe("Biden says US is at tipping point on gun control: We will ban assault weap
20
 
21
  ## Class Labels
22
 
23
- To translate the labels back to subreddit names you need to download the subreddits.json file from this repo manually
24
 
25
  ```py
26
  import json
 
11
  ## Reddit post classification
12
 
13
  This model predicts the subreddit of a provided post
14
+ The transformers library is required
15
+ ```
16
+ pip install 'transformers[torch]'
17
+ ```
18
 
19
  ```py
20
  from transformers import pipeline
 
24
 
25
  ## Class Labels
26
 
27
+ To translate the labels back to subreddit names you need to download the `subreddits.json` file from this repo manually
28
 
29
  ```py
30
  import json