update: load_model

app.py

@@ -63,13 +63,11 @@ lang_codes_by_name = \
  'Galician': 'gl_ES',
  'Slovene': 'sl_SI'}
 
-# @st.cache(allow_output_mutation=True)
+@st.cache(allow_output_mutation=True)
 def load_model():
-    f = open("model.pkl", "rb")
-    return pickle.load(f)
-    #return MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
+    return MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
 
-# @st.cache(allow_output_mutation=True)
+@st.cache(allow_output_mutation=True)
 def load_tokenizer():
     return MBart50TokenizerFast.from_pretrained("facebook/mbart-large-50-many-to-many-mmt", use_fast=False)