Aravindhan-Master commited on
Commit
c88f52d
·
1 Parent(s): dad38f2

[fix]: url path changes

Browse files
Files changed (1) hide show
  1. main.py +3 -3
main.py CHANGED
@@ -13,7 +13,7 @@ class ChunkSimilarityRequest(BaseModel):
13
  model = CrossEncoder('cross-encoder/ms-marco-MiniLM-L6-v2')
14
 
15
 
16
- def cross_encoder_similarity(query, candidates):
17
  pairs = [[query, candidate] for candidate in candidates]
18
  scores = model.predict(pairs)
19
  return scores
@@ -21,7 +21,7 @@ def cross_encoder_similarity(query, candidates):
21
 
22
 
23
  @app.get("/")
24
- async def cross_encoding_similarity():
25
  return {"status": "success"}
26
 
27
 
@@ -29,7 +29,7 @@ async def cross_encoding_similarity():
29
  async def cross_encoding_similarity(
30
  payload: ChunkSimilarityRequest
31
  ):
32
- scores = cross_encoder_similarity(payload.query, payload.chunks)
33
 
34
  return [
35
  chunk for chunk, _ in sorted(zip(payload.chunks, scores), key=lambda x: x[1], reverse=True)[:payload.top_k]
 
13
  model = CrossEncoder('cross-encoder/ms-marco-MiniLM-L6-v2')
14
 
15
 
16
+ def keyword_similarity(query, candidates):
17
  pairs = [[query, candidate] for candidate in candidates]
18
  scores = model.predict(pairs)
19
  return scores
 
21
 
22
 
23
  @app.get("/")
24
+ async def health():
25
  return {"status": "success"}
26
 
27
 
 
29
  async def cross_encoding_similarity(
30
  payload: ChunkSimilarityRequest
31
  ):
32
+ scores = keyword_similarity(payload.query, payload.chunks)
33
 
34
  return [
35
  chunk for chunk, _ in sorted(zip(payload.chunks, scores), key=lambda x: x[1], reverse=True)[:payload.top_k]