added main.py, do not run

This commit is contained in:
henri123lemoine
2023-02-06 20:52:21 -05:00
parent 59d9f25f65
commit 82e79354f4
+13
View File
@@ -0,0 +1,13 @@
from dataset import Dataset
from semantic_search import AlignmentSearch
from settings import DATA_PATH
def main():
dataset = Dataset(DATA_PATH)
dataset.get_alignment_texts()
dataset.load_embeddings()
search_and_answer = AlignmentSearch(dataset)
question = "What is an agent?"
answer = search_and_answer.search_and_answer(question)
print(answer)