Update app.py

app.py

@@ -27,12 +27,6 @@ def searchEmbeddings(id):
 
     image_index = faiss.IndexFlatIP(dim)
 
-    # load dictionaries
-    with open("id_emb_dict.pickle", "rb") as f:
-        id_to_emb_dict = pickle.load(f)
-    with open("indx_to_id.pickle", "rb") as f:
-        indx_to_id_dict = pickle.load(f)
-
     # get index
     image_index = faiss.read_index("image_index.index")
 
@@ -86,6 +80,10 @@ with gr.Blocks() as demo:
         dataset_image1 = pickle.load(f)
     with open("dataset_image2.pickle", "rb") as f:
         dataset_image2 = pickle.load(f)
+    with open("id_emb_dict.pickle", "rb") as f:
+        id_to_emb_dict = pickle.load(f)
+    with open("indx_to_id.pickle", "rb") as f:
+        indx_to_id_dict = pickle.load(f)
         
     with gr.Column():
         process_id = gr.Textbox(label="ID:", info="Enter a sample ID to search for")