Update main.py

main.py

@@ -1,16 +1,26 @@
 from fastapi import FastAPI, HTTPException
-from pydantic import BaseModel
+from pydantic import BaseModel, Field
 from sentence_transformers import SentenceTransformer
 import uvicorn
+import asyncio
+from concurrent.futures import ThreadPoolExecutor
 
-app = FastAPI(title="Text Embedding API (multilingual‑e5‑small)")
+app = FastAPI(title="Text Embedding API (multilingual-e5-small)")
 
 class TextRequest(BaseModel):
-    text: str
+    text: str = Field(..., min_length=1, max_length=10000, description="Text to embed")
 
-# Load Model
+# Globals
+model = None
 model_id = 'intfloat/multilingual-e5-small'
-model = SentenceTransformer(model_id)
+executor = ThreadPoolExecutor(max_workers=4)
+
+@app.on_event("startup")
+async def load_model():
+    global model
+    print(f"Loading model: {model_id}...")
+    model = SentenceTransformer(model_id)
+    print("Model loaded successfully")
 
 @app.get("/")
 def home():
@@ -19,17 +29,20 @@ def home():
 @app.post("/embed/text")
 async def embed_text(request: TextRequest):
     try:
-        # Generate embedding
-        embedding = model.encode(request.text).tolist()
+        loop = asyncio.get_event_loop()
+        embedding = await loop.run_in_executor(
+            executor,
+            lambda: model.encode(request.text, normalize_embeddings=True).tolist()
+        )
         
         return {
-            "success": True, 
+            "success": True,
             "model": model_id,
-            "dimension": len(embedding), 
+            "dimension": len(embedding),
             "embedding": embedding
         }
     except Exception as e:
-        raise HTTPException(status_code=400, detail=str(e))
+        raise HTTPException(status_code=500, detail=str(e))
 
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)