Upload app.py

app.py

@@ -7,6 +7,12 @@ from fastapi import FastAPI, Depends
 from fastapi_health import health
 import logging
 import sys
+from pydantic import BaseModel
+
+class TextPayload(BaseModel):
+    text: str
+
+
 logger = logging.getLogger(__name__)
 
 logging.basicConfig(
@@ -121,9 +127,11 @@ model = EmbeddingModels()
 logging.info('Model Loaded!')
 
 @app.post("/embed-text-gb/")
-async def embed_text(text: str = Query(...)):
+async def embed_text(payload: TextPayload):
     try:
-        embeddings = model.get_text_embedding(text)
+        # Extract the text from the payload
+        embeddings = model.get_text_embedding(payload.text)
         return embeddings
     except Exception as e:
-        logging.info(f'Error: {e}')
+        logging.info(f'Error: {e}')
+        return {"error": str(e)}