Update app.py

app.py

@@ -6,7 +6,7 @@ import torch
 
 # Define the API schema for the request body
 class EmbeddingRequest(BaseModel):
-    text: list[str]
+    inputs: list[str]
 
 # Initialize FastAPI app
 app = FastAPI()
@@ -21,7 +21,7 @@ model = SentenceTransformer(model_name, device=device, trust_remote_code=True)
 async def get_embeddings(request: EmbeddingRequest):
     try:
         # Get embeddings for the input text
-        embeddings = model.encode(request.text, convert_to_numpy=True).tolist()
+        embeddings = model.encode(request.inputs, convert_to_numpy=True).tolist()
         return {"embeddings": embeddings}
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))