Added cache_resource functionality to speed up the space

app.py

@@ -9,6 +9,10 @@ import io
 
 st.set_page_config(page_title="Translation Demo", page_icon=":milky_way:", layout="wide")
 
+@st.cache_resource
+def load_model():
+    return M2M100ForConditionalGeneration.from_pretrained("alirezamsh/small100")
+
 
 def get_translation(src_code, trg_code, src):
 
@@ -98,7 +102,7 @@ target = ""
 #model = MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
 #tokenizer = MBart50TokenizerFast.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
 
-model = M2M100ForConditionalGeneration.from_pretrained("alirezamsh/small100")
+
 tokenizer = SMALL100Tokenizer.from_pretrained("alirezamsh/small100")
 
 #valid_languages = ['de_DE', 'en_XX', 'it_IT']