Update main.py

main.py

@@ -5,7 +5,7 @@ from typing import List
 import os, platform
 
 
-model = AutoModel.from_pretrained('jinaai/jina-embeddings-v2-base-en', trust_remote_code=True)
+model = AutoModel.from_pretrained('moraxgiga/jina', trust_remote_code=True)
 
 app = FastAPI()
 
@@ -15,11 +15,5 @@ class Validation(BaseModel):
 
 #Endpoint
 @app.post("/jina_embedding")
-async def chaatie_agent(item: Validation):
-    # Assuming model.encode returns a list of numpy arrays (one for each prompt)
-    embeddings = model.encode(item.prompt)
-
-    # Convert each numpy array in the list to a list
-    embeddings_list = [embedding.tolist() for embedding in embeddings]
-
-    return {"embeddings": embeddings_list}
+async def generate_embeddings(text):
+    return {"embeddings": model.encode(text).tolist()}