Spaces:
Runtime error
Runtime error
upgrade from distilbert to bert
Browse files
app.py
CHANGED
@@ -58,7 +58,7 @@ def predict(text, tokenizer, model, temperature = 1):
|
|
58 |
|
59 |
return result
|
60 |
|
61 |
-
model, tokenizer = get_model('
|
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")
|