Update app.py

app.py

@@ -6,8 +6,8 @@ import os
 import torch
 from transformers import RobertaModel, AutoTokenizer
 
-
-@st.cache(show_spinner=False, hash_funcs={RobertaModel: lambda _: None, AutoTokenizer: lambda _: None, dict: lambda _: None})
+# @st.cache(hash_funcs={transformers.models.gpt2.tokenization_gpt2_fast.GPT2TokenizerFast: hash}, suppress_st_warning=True, allow_output_mutation=True)
+@st.cache(show_spinner=False, hash_funcs={RobertaModel: hash, AutoTokenizer: hash})
 def load():
     text_encoder = RobertaModel.from_pretrained('SajjadAyoubi/clip-fa-text')
     tokenizer = AutoTokenizer.from_pretrained('SajjadAyoubi/clip-fa-text')