wbrooks commited on
Commit
f6d14bf
·
1 Parent(s): 68239c7

fixed a typo and added a test endpoint

Browse files
Files changed (2) hide show
  1. .gitattributes +1 -0
  2. app.py +9 -1
.gitattributes CHANGED
@@ -37,3 +37,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
37
  pixi.lock merge=binary linguist-language=YAML linguist-generated=true
38
  outputs/block_embeddings_df.parquet.zstd filter=lfs diff=lfs merge=lfs -text
39
  outputs/TF-IDF-doc-text.csv filter=lfs diff=lfs merge=lfs -text
 
 
37
  pixi.lock merge=binary linguist-language=YAML linguist-generated=true
38
  outputs/block_embeddings_df.parquet.zstd filter=lfs diff=lfs merge=lfs -text
39
  outputs/TF-IDF-doc-text.csv filter=lfs diff=lfs merge=lfs -text
40
+ outputs/dtm_svd.joblib filter=lfs diff=lfs merge=lfs -text
app.py CHANGED
@@ -9,7 +9,7 @@ app = FastAPI()
9
  @app.get("/search")
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
 
@@ -21,3 +21,11 @@ def greet_json(query: str):
21
  return {"result": str(res_combined[:10,])}
22
 
23
 
 
 
 
 
 
 
 
 
 
9
  @app.get("/search")
10
  def greet_json(query: str):
11
  res_tfidf = query_docs(query)
12
+ res_sbert = sbert_query_docs(query)
13
 
14
  joined = res_sbert.join(res_tfidf, on='file', how = 'inner')
15
 
 
21
  return {"result": str(res_combined[:10,])}
22
 
23
 
24
+
25
+
26
+ @app.get("/test")
27
+ def echo(query: str):
28
+
29
+ return {"echo": query}
30
+
31
+