sam2ai commited on
Commit
c61b22c
·
1 Parent(s): f1e422e

Synced repo using 'sync_with_huggingface' Github Action

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -83,12 +83,12 @@ app = FastAPI(title="embedding Inference")
83
 
84
  @app.post("/embed")
85
  async def api_embed(
86
- textA: str = Body("text1", description="", embed=True),
87
- textB: str = Body("text2", description="", embed=True),
88
  ):
89
 
90
- q_embeddings = model.encode(textA, normalize_embeddings=True)
91
- p_embeddings = model.encode(textB, normalize_embeddings=True)
92
 
93
  scores = q_embeddings @ p_embeddings.T
94
  print(scores)
 
83
 
84
  @app.post("/embed")
85
  async def api_embed(
86
+ text1: str = Body("text1", description="", embed=True),
87
+ text2: str = Body("text2", description="", embed=True),
88
  ):
89
 
90
+ q_embeddings = model.encode(text1, normalize_embeddings=True)
91
+ p_embeddings = model.encode(text2, normalize_embeddings=True)
92
 
93
  scores = q_embeddings @ p_embeddings.T
94
  print(scores)