Merge pull request #46 from deepankarm/main

fix(windows): move global recommender
pull/50/head
Bhaskar Tripathi 1 year ago committed by GitHub
commit 4c240ae9e2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -14,6 +14,9 @@ from lcserve import serving
from sklearn.neighbors import NearestNeighbors
recommender = None
def download_pdf(url, output_path):
urllib.request.urlretrieve(url, output_path)
@ -97,6 +100,9 @@ class SemanticSearch:
def load_recommender(path, start_page=1):
global recommender
if recommender is None:
recommender = SemanticSearch()
texts = pdf_to_text(path, start_page=start_page)
chunks = text_to_chunks(texts, start_page=start_page)
recommender.fit(chunks)
@ -140,9 +146,6 @@ def generate_answer(question, openAI_key):
return answer
recommender = SemanticSearch()
def load_openai_key() -> str:
key = os.environ.get("OPENAI_API_KEY")
if key is None:

Loading…
Cancel
Save