dreji18 commited on
Commit
92ba70e
1 Parent(s): 1f4e0d9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -30,8 +30,8 @@ The easiest way is to load through the pipeline object offered by transformers l
30
  ```python
31
  from transformers import AutoTokenizer, TFAutoModelForSequenceClassification
32
  from transformers import pipeline
33
- tokenizer = AutoTokenizer.from_pretrained("dreji18/bias-detection-model", use_auth_token=True)
34
- model = TFAutoModelForSequenceClassification.from_pretrained("dreji18/bias-detection-model", use_auth_token=True)
35
 
36
  classifier = pipeline('text-classification', model=model, tokenizer=tokenizer) # cuda = 0,1 based on gpu availability
37
  classifier("The irony, of course, is that the exhibit that invites people to throw trash at vacuuming Ivanka Trump lookalike reflects every stereotype feminists claim to stand against, oversexualizing Ivanka’s body and ignoring her hard work.")
 
30
  ```python
31
  from transformers import AutoTokenizer, TFAutoModelForSequenceClassification
32
  from transformers import pipeline
33
+ tokenizer = AutoTokenizer.from_pretrained("d4data/bias-detection-model", use_auth_token=True)
34
+ model = TFAutoModelForSequenceClassification.from_pretrained("d4data/bias-detection-model", use_auth_token=True)
35
 
36
  classifier = pipeline('text-classification', model=model, tokenizer=tokenizer) # cuda = 0,1 based on gpu availability
37
  classifier("The irony, of course, is that the exhibit that invites people to throw trash at vacuuming Ivanka Trump lookalike reflects every stereotype feminists claim to stand against, oversexualizing Ivanka’s body and ignoring her hard work.")