Update main.py

main.py

@@ -16,5 +16,5 @@ class Validation(BaseModel):
 
 #Endpoint
 @app.post("/jina_embedding")
-async def generate_embeddings(text):
-    return {"embeddings": model.encode(text).tolist()}
+async def generate_embeddings(item: Validation):
+    return {"embeddings": model.encode(item.prompt).tolist()}