Update app.py

app.py

@@ -1,20 +1,9 @@
 import streamlit as st
-import pandas as pd
 import numpy as np
-from time import time
 from html import escape
-import os
 import torch
-import transformers
 from transformers import RobertaModel, AutoTokenizer
 
-#@st.cache(show_spinner=False)
-#def load():
-#    text_encoder = RobertaModel.from_pretrained('SajjadAyoubi/clip-fa-text')
-#    image_embeddings = torch.load('embedding.pt')
-#    links = np.load('data.npy', allow_pickle=True)
-#    return text_encoder, links, image_embeddings
-
 
 tokenizer = AutoTokenizer.from_pretrained('SajjadAyoubi/clip-fa-text')
 text_encoder = RobertaModel.from_pretrained('SajjadAyoubi/clip-fa-text').eval()
@@ -23,10 +12,10 @@ links = np.load('data.npy', allow_pickle=True)
 
 
 
-def get_html(url_list, height=200):
+def get_html(url_list):
     html = "<div style='margin-top: 50px; max-width: 1200px; display: flex; flex-wrap: wrap; justify-content: space-evenly'>"
     for url in url_list:
-        html2 = f"<img style='height: {height}px; margin: 2px' src='{escape(url)}'>"  
+        html2 = f"<img style='height: 200px; margin: 2px' src='{escape(url)}'>"  
         html = html + html2
     html += "</div>"
     return html
@@ -50,15 +39,6 @@ def main():
               .block-container{
                 max-width: 1200px;
               }
-              div.row-widget.stRadio > div{
-                flex-direction:row;
-                display: flex;
-                justify-content: center;
-              }
-              div.row-widget.stRadio > div > label{
-                margin-left: 5px;
-                margin-right: 5px;
-              }
               section.main>div:first-child {
                 padding-top: 0px;
               }
@@ -76,20 +56,14 @@ def main():
               }
               </style>''',
                 unsafe_allow_html=True)
-    st.sidebar.markdown(description)
+                
+    # st.sidebar.markdown(description)
     _, c, _ = st.columns((1, 3, 1))
     query = c.text_input('Search text', value='مرغ دریای')
     if len(query) > 0:
         results = image_search(query)
-        start = time()
         st.markdown(get_html(results), unsafe_allow_html=True)
-        end = time()
-        st.text(str(end - start))
-
-
-
 
-print()
 
 if __name__ == '__main__':
     main()