Update main.py

main.py

@@ -15,6 +15,15 @@ class Validation(BaseModel):
 
 
 #Endpoint
-@app.post("/jina_embedding")
+@@app.post("/jina_embedding")
 async def generate_embeddings(item: Validation):
-    return {"embeddings": model.encode(item.prompt).tolist()}
+    start_time = time.time()  
+    embeddings = model.encode(item.prompt).tolist()
+    end_time = time.time()  
+    time_taken = end_time - start_time  # Calculate the time taken
+
+    return {
+        "embeddings": embeddings,
+        "time_taken": f"{time_taken:.2f} seconds",
+        "Number_of_sentence_processed" : len(item.prompt)
+    }