Update app.py

app.py

@@ -5,10 +5,7 @@ import torch
 from transformers import RobertaModel, AutoTokenizer
 
 
-@st.cache(show_spinner=False,
-          hash_funcs={AutoTokenizer: lambda _: None,
-                      RobertaModel: lambda _: None,
-                      dict: lambda _: None})
+@st.experimental_memo
 def load():
   tokenizer = AutoTokenizer.from_pretrained('SajjadAyoubi/clip-fa-text')
   text_encoder = RobertaModel.from_pretrained('SajjadAyoubi/clip-fa-text').eval()
@@ -28,7 +25,7 @@ def get_html(url_list):
     html += "</div>"
     return html
  
-st.cache(show_spinner=False)                           
+ 
 def image_search(query, top_k=10):
     with torch.no_grad():
         text_embedding = text_encoder(**tokenizer(query, return_tensors='pt')).pooler_output