Fixed load_model() method

app.py

@@ -11,8 +11,8 @@ st.set_page_config(page_title="Translation Demo", page_icon=":milky_way:", layou
 
 @st.cache
 def load_model():
-    return M2M100ForConditionalGeneration.from_pretrained("alirezamsh/small100")
-
+    model = M2M100ForConditionalGeneration.from_pretrained("alirezamsh/small100")
+    return model
 
 def get_translation(src_code, trg_code, src):
 
@@ -23,6 +23,7 @@ def get_translation(src_code, trg_code, src):
         #forced_bos_token_id=tokenizer.lang_code_to_id[trg_code]
     #)
     #trg = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
+    model = load_model()
     tokenizer.tgt_lang = trg_code
     encoded = tokenizer(src, return_tensors="pt")
     generated_tokens = model.generate(**encoded)