From 82e79354f4e92fa65b7af6b27ee442353f67aa75 Mon Sep 17 00:00:00 2001 From: henri123lemoine Date: Mon, 6 Feb 2023 20:52:21 -0500 Subject: [PATCH] added main.py, do not run --- src/main.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) create mode 100644 src/main.py diff --git a/src/main.py b/src/main.py new file mode 100644 index 0000000..c0e77e1 --- /dev/null +++ b/src/main.py @@ -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) \ No newline at end of file