Upload classifier.py

classifier.py

@@ -15,13 +15,13 @@ COLOURS = {
 @st.cache(allow_output_mutation=True)
 def load_model():
     print('Loading classification model')
-    return BertForTokenClassification.from_pretrained("classifier/")
+    return BertForTokenClassification.from_pretrained("aligator/mBERT_grammatical_error_tagger")
 
 
 @st.cache(allow_output_mutation=True)
 def load_tokenizer():
     print("Loading tokenizer for classification model")
-    return BertTokenizerFast.from_pretrained("classifier/")
+    return BertTokenizerFast.from_pretrained("aligator/mBERT_grammatical_error_tagger")
 
 
 model = load_model()