Update app.py

app.py

@@ -19,7 +19,7 @@ suggested_text_list = ['מטרת הבריאה']
 @st.cache(hash_funcs={tokenizers.Tokenizer: id, tokenizers.AddedToken: id})
 def load_model(model_name):
     tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModelForCausalLM.from_pretrained(model_name)
+    model = AutoModelForCausalLM.from_pretrained(model_name, low_cpu_mem_usage=True)
     return model, tokenizer
 
 def extend(input_text, max_size=20, top_k=50, top_p=0.95):