Update app.py

app.py

@@ -1,9 +1,19 @@
 from clip_multilingual.search import MultiLingualSearch
 
-#images_embeddings = np.load('/path/to/images_embeddings')
-#images_data = [...] # List of image info for each row of the embeddings. For instance, it could be a list of urls, filepaths, ids. They will be returned when calling the search function
+@st.cache(
+    suppress_st_warning=True,
+    hash_funcs={
+        Tokenizer: lambda _: None
+    }
+)
+def load_model():
+    unsplash_base_folder = './'
+    all_embeddings = np.load(f'{unsplash_base_folder}/embeddings.npy')
+    with open(f'{unsplash_base_folder}/urls.json') as f:
+        all_urls = json.load(f)
+    return MultiLingualSearch(all_embeddings, all_urls)
 
-model = None
-semantic_search = MultiLingualSearch(model, images_embeddings, images_data)
+semantic_search = load_model()
 
 results = semantic_search.search('people smiling')
+print(result)