Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -27,18 +27,26 @@ app = FastAPI(
|
|
| 27 |
class TextRequest(BaseModel):
|
| 28 |
text: str
|
| 29 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 30 |
@app.get("/health")
|
| 31 |
def health():
|
| 32 |
return {"status": "ok"}
|
| 33 |
|
| 34 |
@app.post("/embed")
|
| 35 |
def embed(req: TextRequest):
|
|
|
|
|
|
|
|
|
|
| 36 |
with torch.no_grad():
|
| 37 |
embedding = model.encode(
|
| 38 |
req.text,
|
| 39 |
convert_to_numpy=True,
|
| 40 |
normalize_embeddings=True
|
| 41 |
)
|
|
|
|
| 42 |
return {
|
| 43 |
"embedding": embedding.tolist(),
|
| 44 |
"dim": len(embedding)
|
|
|
|
| 27 |
class TextRequest(BaseModel):
|
| 28 |
text: str
|
| 29 |
|
| 30 |
+
@app.get("/")
|
| 31 |
+
def root():
|
| 32 |
+
return {"message": "API is running"}
|
| 33 |
+
|
| 34 |
@app.get("/health")
|
| 35 |
def health():
|
| 36 |
return {"status": "ok"}
|
| 37 |
|
| 38 |
@app.post("/embed")
|
| 39 |
def embed(req: TextRequest):
|
| 40 |
+
if model is None:
|
| 41 |
+
return {"error": "Model not loaded"}
|
| 42 |
+
|
| 43 |
with torch.no_grad():
|
| 44 |
embedding = model.encode(
|
| 45 |
req.text,
|
| 46 |
convert_to_numpy=True,
|
| 47 |
normalize_embeddings=True
|
| 48 |
)
|
| 49 |
+
|
| 50 |
return {
|
| 51 |
"embedding": embedding.tolist(),
|
| 52 |
"dim": len(embedding)
|