sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2024-11-23 15:38:55 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

Create script to install semantic-search as a program

Browse source
This commit is contained in:
Debanjum Singh Solanky 2021-08-02 00:29:09 -07:00
parent 13d5100ce6
commit 2eb029a7b0
1 changed files with 25 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
install.py Normal file
View file

@ -0,0 +1,25 @@
#!/usr/bin/env python3
import pathlib
import argparse
if __name__ == '__main__':
# Setup Argument Parser
parser = argparse.ArgumentParser(description="Setup the semantic search program")
parser.add_argument('--script-dir', '-s', default="./", type=pathlib.Path, help="The project directory. Default: Current Directory")
parser.add_argument('--install-path', '-i', default="/usr/bin/semantic-search", type=pathlib.Path, help="The directory to install the script. Default: ./embeddings.pt")
parser.add_argument('--model-dir', '-m', default="./", type=pathlib.Path, help="The directory to store the model in. Default: Current Directory")
args = parser.parse_args()
run_script_content = f'''#!/bin/bash
# Arrange
eval "$(conda shell.bash hook)"
conda activate samvayati
cd {args.script_dir.expanduser().absolute()}
# Act
python3 asymmetric.py -j {args.model_dir.expanduser().absolute()}/notes.jsonl.gz -e {args.model_dir.expanduser().absolute()}/notes_embeddings.pt -n 5 --interactive
'''
with args.install_path.open(mode='w') as run_script:
run_script.write(run_script_content)