ishmeet-yo commited on
Commit
cc8a7d4
·
verified ·
1 Parent(s): 6194f24

Update app/main.py

Browse files
Files changed (1) hide show
  1. app/main.py +2 -2
app/main.py CHANGED
@@ -11,7 +11,7 @@ templates = Jinja2Templates(directory="templates")
11
  app.mount("/static", StaticFiles(directory="static"), name="static")
12
 
13
  # Load your RAG data once
14
- documents, embeddings, model = load_data()
15
 
16
  @app.get("/", response_class=HTMLResponse)
17
  async def home(request: Request):
@@ -22,7 +22,7 @@ async def home(request: Request):
22
 
23
  @app.post("/", response_class=HTMLResponse)
24
  async def ask(request: Request, query: str = Form(...)):
25
- results = retrieve_chunks(query, documents, embeddings, model)
26
  answer = results[0] if results else "No relevant answer found."
27
 
28
  return templates.TemplateResponse(
 
11
  app.mount("/static", StaticFiles(directory="static"), name="static")
12
 
13
  # Load your RAG data once
14
+ documents, embeddings = load_data()
15
 
16
  @app.get("/", response_class=HTMLResponse)
17
  async def home(request: Request):
 
22
 
23
  @app.post("/", response_class=HTMLResponse)
24
  async def ask(request: Request, query: str = Form(...)):
25
+ results = retrieve_chunks(query, documents, embeddings)
26
  answer = results[0] if results else "No relevant answer found."
27
 
28
  return templates.TemplateResponse(