ethical-spectacle
/

social-bias-ner

Token Classification

Inference Endpoints

Model card Files Files and versions Community

maximuspowers commited on Sep 20

Commit

3d45c2a

•

1 Parent(s): ca4a7b0

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -44,7 +44,7 @@ import gradio as gr
 # init important things
 tokenizer = BertTokenizerFast.from_pretrained('bert-base-uncased')
-model = BertForTokenClassification.from_pretrained('maximuspowers/bias-detection-ner')
 model.eval()
 model.to('cuda' if torch.cuda.is_available() else 'cpu')

 # init important things
 tokenizer = BertTokenizerFast.from_pretrained('bert-base-uncased')
+model = BertForTokenClassification.from_pretrained('ethical-spectacle/social-bias-ner')
 model.eval()
 model.to('cuda' if torch.cuda.is_available() else 'cpu')