atrytone commited on
Commit
ab84f0a
1 Parent(s): 7721b02

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -3,8 +3,9 @@ from transformers import AutoModelForSequenceClassification
3
  import gradio as gr
4
 
5
  model_name = 'biodatlab/score-claim-identification'
 
6
 
7
- tokenizer = AutoTokenizer.from_pretrained(model_name)
8
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
9
 
10
  def inference(abstract):
 
3
  import gradio as gr
4
 
5
  model_name = 'biodatlab/score-claim-identification'
6
+ tokenizer_name = 'allenai/scibert_scivocab_uncased'
7
 
8
+ tokenizer = AutoTokenizer.from_pretrained(tokenizer_name)
9
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
10
 
11
  def inference(abstract):