Hugging Face's logo

Spaces:
maximuspowers
/
bias-detection-ner
Running

App Files Community
maximuspowers commited on
Commit
9717ed1
·
verified ·
1 Parent(s): 9e4d863

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import gradio as gr
4
 
5
  # Load tokenizer and model
6
  tokenizer = BertTokenizerFast.from_pretrained('bert-base-uncased')
7
- model = BertForTokenClassification.from_pretrained('./saved_model3')
8
  model.eval()
9
  model.to('cuda' if torch.cuda.is_available() else 'cpu')
10
 
 
4
 
5
  # Load tokenizer and model
6
  tokenizer = BertTokenizerFast.from_pretrained('bert-base-uncased')
7
+ model = BertForTokenClassification.from_pretrained('maximuspowers/bias-detection-ner')
8
  model.eval()
9
  model.to('cuda' if torch.cuda.is_available() else 'cpu')
10