Synced repo using 'sync_with_huggingface' Github Action
Browse files
app.py
CHANGED
|
@@ -83,12 +83,12 @@ app = FastAPI(title="embedding Inference")
|
|
| 83 |
|
| 84 |
@app.post("/embed")
|
| 85 |
async def api_embed(
|
| 86 |
-
|
| 87 |
-
|
| 88 |
):
|
| 89 |
|
| 90 |
-
q_embeddings = model.encode(
|
| 91 |
-
p_embeddings = model.encode(
|
| 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)
|