Update app.py

app.py

@@ -4,16 +4,10 @@ from html import escape
 import torch
 from transformers import RobertaModel, AutoTokenizer
 
-
-@st.experimental_memo
-def load():
-  tokenizer = AutoTokenizer.from_pretrained('SajjadAyoubi/clip-fa-text')
-  text_encoder = RobertaModel.from_pretrained('SajjadAyoubi/clip-fa-text').eval()
-  image_embeddings = torch.load('embedding.pt')
-  links = np.load('data.npy', allow_pickle=True)
-  return tokenizer, text_encoder, links, image_embeddings
-  
-tokenizer, text_encoder, links, image_embeddings = load()
+tokenizer = AutoTokenizer.from_pretrained('SajjadAyoubi/clip-fa-text')
+text_encoder = RobertaModel.from_pretrained('SajjadAyoubi/clip-fa-text').eval()
+image_embeddings = torch.load('embedding.pt')
+links = np.load('data.npy', allow_pickle=True)
 
 
 @st.experimental_memo
@@ -27,7 +21,7 @@ def image_search(query, top_k=10):
 def get_html(url_list):
     html = "<div style='margin-top: 50px; max-width: 1100px; display: flex; flex-wrap: wrap; justify-content: space-evenly'>"
     for url in url_list:
-        html2 = f"<img style='height: 200px; margin: 2px' src='{escape(url)}'>"  
+        html2 = f"<img style='height: 180px; margin: 2px' src='{escape(url)}'>"  
         html = html + html2
     html += "</div>"
     return html