Update app.py

app.py

@@ -2,13 +2,21 @@ from fastapi import FastAPI
 from pydantic import BaseModel
 from sentence_transformers import SentenceTransformer
 
+# Initialize FastAPI app
 app = FastAPI()
+
+# Load the BGE Chinese model
 model = SentenceTransformer("BAAI/bge-small-zh")
 
+# Define request format
 class Texts(BaseModel):
     texts: list[str]
 
+# Embed endpoint
 @app.post("/embed")
-async def embed(input: Texts):
-    embeddings = model.encode(input.texts, normalize_embeddings=True).tolist()
+async def embed(texts: Texts):
+    embeddings = model.encode(
+        texts.texts,
+        normalize_embeddings=True
+    ).tolist()
     return {"embeddings": embeddings}