Upload corrector.py

update huggingface name

corrector.py

@@ -9,13 +9,13 @@ s3 = S3FileSystem(anon=True)
 @st.cache(allow_output_mutation=True)
 def load_model():
     print("Load correction model")
-    return MBartForConditionalGeneration.from_pretrained("aligator/mBART_french_correction")
+    return MBartForConditionalGeneration.from_pretrained("alice-hml/mBART_french_correction")
 
 
 @st.cache(allow_output_mutation=True)
 def load_tokenizer():
     print("Load tokenizer for correction model")
-    return MBartTokenizerFast.from_pretrained("aligator/mBART_french_correction")
+    return MBartTokenizerFast.from_pretrained("alice-hml/mBART_french_correction")
 
 
 model = load_model()