Spaces:

SaviAnna
/

history_mistery

Sleeping

SaviAnna commited on Jun 9, 2023

Commit

d139c80

1 Parent(s): ec78c6b

Update pages/📖History_Mystery.py

Files changed (1) hide show

pages/📖History_Mystery.py CHANGED Viewed

@@ -14,7 +14,7 @@ max_length = st.slider(" Длина сгенерированного отрыв
 # model = GPT2LMHeadModel.from_pretrained('sberbank-ai/rugpt3small_based_on_gpt2')
 # tokenizer = GPT2Tokenizer.from_pretrained('sberbank-ai/rugpt3small_based_on_gpt2')
 # #Задаем класс модели (уже в streamlit/tg_bot)
-@st.cache(allow_output_mutation=True)
 def load_gpt():
     model_GPT = GPT2LMHeadModel.from_pretrained(
      'sberbank-ai/rugpt3small_based_on_gpt2',

 # model = GPT2LMHeadModel.from_pretrained('sberbank-ai/rugpt3small_based_on_gpt2')
 # tokenizer = GPT2Tokenizer.from_pretrained('sberbank-ai/rugpt3small_based_on_gpt2')
 # #Задаем класс модели (уже в streamlit/tg_bot)
+@st.cache_data(allow_output_mutation=True)
 def load_gpt():
     model_GPT = GPT2LMHeadModel.from_pretrained(
      'sberbank-ai/rugpt3small_based_on_gpt2',