Update app.py

app.py

@@ -110,9 +110,12 @@ with gr.Blocks() as demo:
         id_to_dna_emb_dict = pickle.load(f)
 
     # open taxonomy files
-    family = [item.decode("utf-8") for item in dataset_hdf5_all_key["family"][:]]
-    genus = [item.decode("utf-8") for item in dataset_hdf5_all_key["genus"][:]]
-    species = [item.decode("utf-8") for item in dataset_hdf5_all_key["species"][:]]
+    with open("family.pickle", "rb") as f:
+        family = [item.decode("utf-8") for item in pickle.load(f)]
+    with open("genus.pickle", "rb") as f:
+        genus= [item.decode("utf-8") for item in pickle.load(f)]
+    with open("species.pickle", "rb") as f:
+        species = [item.decode("utf-8") for item in pickle.load(f)]
 
     with gr.Column():
         process_id = gr.Textbox(label="ID:", info="Enter a sample ID to search for")