mirror of
https://github.com/wassname/stampy-chat.git
synced 2026-10-30 13:50:18 +08:00
added main.py, do not run
This commit is contained in:
1 parent
59d9f25f65
commit
82e79354f4
1 file changed
+13
+13
@@ -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)
|
||||
Reference in new issue
Block a user