vladyur commited on
Commit
9001366
1 Parent(s): 95c7eb3

upgrade from distilbert to bert

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -58,7 +58,7 @@ def predict(text, tokenizer, model, temperature = 1):
58
 
59
  return result
60
 
61
- model, tokenizer = get_model('distilbert-base-cased', 'distilbert-checkpoint-10983.bin')
62
 
63
  st.title("Yandex School of Data Analysis. ML course")
64
  st.title("Laboratory work 2: classifier of categories of scientific papers")
 
58
 
59
  return result
60
 
61
+ model, tokenizer = get_model('bert-base-cased', 'bert-checkpoint-14644.bin')
62
 
63
  st.title("Yandex School of Data Analysis. ML course")
64
  st.title("Laboratory work 2: classifier of categories of scientific papers")