Roar / main.py
rohan13's picture
Roar code
b8b8495
raw
history blame contribute delete
382 Bytes
from utils import get_search_index, generate_answer, set_model_and_embeddings, ingest
def index(model, first_time=False):
set_model_and_embeddings(model)
get_search_index(model, first_time=first_time)
return True
def ingest_files(file_paths, model):
return ingest(file_paths, model)
def run(question, model):
index(model)
return generate_answer(question)