Update main.py

main.py

@@ -31,6 +31,7 @@ def chunk_text(text, chunk_size=512):
 @app.post("/get_embeding")
 async def get_embeding(text):
     chunks = chunk_text(text)
+    all_embeddings = []
     for chunk in chunks:
     
         # Tokenize the input text
@@ -47,8 +48,8 @@ async def get_embeding(text):
         sentence_embedding = torch.mean(embeddings, dim=1)
     
         #print(sentence_embedding)
-    
-        return sentence_embedding
+        all_embeddings.append(sentence_embedding)
+    return all_embeddings