Update app.py

app.py

@@ -7,7 +7,6 @@ import torch
 import transformers
 from transformers import RobertaModel, AutoTokenizer
 
-#@st.cache(hash_funcs={transformers.models.roberta.tokenization_roberta_fast.RobertaTokenizerFast: hash}, suppress_st_warning=True, allow_output_mutation=True)
 @st.cache(show_spinner=False)
 def load():
     text_encoder = RobertaModel.from_pretrained('SajjadAyoubi/clip-fa-text')
@@ -28,7 +27,7 @@ def get_html(url_list, height=224):
     html += "</div>"
     return html
         
-        
+@st.cache(show_spinner=False)        
 def image_search(query, top_k=8):
     with torch.no_grad():
         text_embedding = text_encoder(**tokenizer(query, return_tensors='pt')).pooler_output
@@ -75,7 +74,7 @@ def main():
                 unsafe_allow_html=True)
     st.sidebar.markdown(description)
     _, c, _ = st.columns((1, 3, 1))
-    query = c.text_input('', value='غروب خورشید')
+    query = c.text_input('', value='دریا')
     if len(query) > 0:
         results = image_search(query)
         st.markdown(get_html(results), unsafe_allow_html=True)