Spaces:

Norod78
/

FantasyChildrenScifi-hebrew-gpt_neo-small

Sleeping

Norod78 commited on May 4, 2023

Commit

de72a99

1 Parent(s): b6c25db

cache_resource

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ import tokenizers
 random.seed(None)
 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)

 random.seed(None)
 suggested_text_list = ['השד הופיע מול','קאלי שלפה את','פעם אחת לפני שנים רבות', 'הארי פוטר חייך חיוך נבוך', 'ואז הפרתי את כל כללי הטקס כש']
+@st.cache_resource
 def load_model(model_name):
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     model = AutoModelForCausalLM.from_pretrained(model_name)