mirror of
https://github.com/wassname/stampy-chat.git
synced 2026-09-12 13:00:42 +08:00
added main.py, do not run
This commit is contained in:
+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