Update app.py

app.py

@@ -4,14 +4,9 @@ from transformers import pipeline
 from transformers import AutoTokenizer, AutoModelForMaskedLM
 
 tokenizer = AutoTokenizer.from_pretrained("moussaKam/AraBART", padding= True, truncation=True, max_length=128)
-@st.cache
-def load_model(model_name):
-    model = AutoModelForMaskedLM.from_pretrained(model_name)
-    return model
-model = load_model("moussaKam/AraBART")
+model = AutoModelForMaskedLM.from_pretrained(model_name)
 
 
-@st.cache
 def next_word(text, pipe):
     res_dict= {  
   'token_str':[],