penscola commited on
Commit
6c166a2
·
1 Parent(s): acacb0e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ from scipy.special import softmax
10
  #import the model
11
  tokenizer = AutoTokenizer.from_pretrained('distilbert-base-uncased')
12
 
13
- model_path = f"pytorch_model.bin"
14
  config = AutoConfig.from_pretrained(model_path)
15
  model = AutoModelForSequenceClassification.from_pretrained(model_path)
16
  #Set the page configs
 
10
  #import the model
11
  tokenizer = AutoTokenizer.from_pretrained('distilbert-base-uncased')
12
 
13
+ model_path = f"config.json"
14
  config = AutoConfig.from_pretrained(model_path)
15
  model = AutoModelForSequenceClassification.from_pretrained(model_path)
16
  #Set the page configs