Spaces:

RushabhShah122000
/

Similarity-Search

Runtime error

RushabhShah122000 commited on Aug 20, 2024

Commit

f5e7a4a

verified ·

1 Parent(s): 69a6242

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -206,7 +206,7 @@ index_file = "faiss-d2-nn_index.pkl"
 metadata_file = "faiss-d2-image_metadata.pkl"
 if not os.path.exists(index_file) or not os.path.exists(metadata_file):
-    root_dir = "Similarity-Search/Dataset2"
     index, image_paths, categories = load_and_index_images(root_dir)
     save_index_and_metadata(index, image_paths, categories, index_file, metadata_file)
 else:

 metadata_file = "faiss-d2-image_metadata.pkl"
 if not os.path.exists(index_file) or not os.path.exists(metadata_file):
+    root_dir = "Dataset2"
     index, image_paths, categories = load_and_index_images(root_dir)
     save_index_and_metadata(index, image_paths, categories, index_file, metadata_file)
 else: