Skip to content

Commit

Permalink
readme and pyproject.toml updated
Browse files Browse the repository at this point in the history
  • Loading branch information
AnirudhPraveen committed Nov 15, 2024
1 parent fdafa69 commit 7eb8e56
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 3 deletions.
7 changes: 4 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -80,8 +80,8 @@ def main():
searcher = AudioSimilaritySearch(index_type=IndexType.FLAT)

# Set up dataset
dataset_dir = Path("~/Documents/data").expanduser()
query_file = Path("~/Documents/RogerMoore_2.wav").expanduser()
dataset_dir = Path("dataset_directory").expanduser()
query_file = Path("query_directory").expanduser()

# Get audio files
audio_files = list(dataset_dir.glob("**/*.wav"))
Expand All @@ -90,7 +90,8 @@ def main():
# Add batch to Index files
#searcher.add_batch(audio_files)

saved_index_dir = Path("./saved_indices/index_20241111-094634").expanduser()
saved_index_dir = Path("./saved_index_folder").expanduser()
# do not include the index.faiss file in the directory

# Load saved index
searcher.load(saved_index_dir)
Expand Down
3 changes: 3 additions & 0 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,9 @@ dependencies = [
]

[project.optional-dependencies]
cpu = [
"faiss-cpu>=1.7.0"
]
dev = [
"pytest>=6.0",
"black>=22.0",
Expand Down

0 comments on commit 7eb8e56

Please sign in to comment.