wbrooks commited on
Commit
e2ee208
·
1 Parent(s): 21ca93f

adding very basic app

Browse files
Files changed (1) hide show
  1. app.py +18 -2
app.py CHANGED
@@ -1,7 +1,23 @@
1
  from fastapi import FastAPI
 
 
 
 
2
 
3
  app = FastAPI()
4
 
5
  @app.get("/")
6
- def greet_json():
7
- return {"Hello": "World!"}
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  from fastapi import FastAPI
2
+ from src.do_pca_on_tfidf import query_docs
3
+ from src.search_embeddings import sbert_query_docs
4
+ import polars as pl
5
+
6
 
7
  app = FastAPI()
8
 
9
  @app.get("/")
10
+ def greet_json(query: str):
11
+ res_tfidf = query_docs(query)
12
+ res_sbert = query_docs(query)
13
+
14
+ joined = res_sbert.join(res_tfidf, on='file', how = 'inner')
15
+
16
+ res_combined = joined.with_columns((0.7 * pl.col("rank-sbert") + 0.3 * pl.col("rank-tfidf")).alias("rank-combined")).sort("rank-combined")
17
+
18
+
19
+
20
+
21
+ return {"result": str(res_combined[:10,])}
22
+
23
+