Update main.py

main.py

@@ -25,7 +25,7 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-def chunk_text(text, chunk_size=10):
+def chunk_text(text, chunk_size=512):
     return [text[i:i + chunk_size] for i in range(0, len(text), chunk_size)]
 
 @app.post("/get_embeding")
@@ -48,7 +48,7 @@ async def get_embeding(text):
         sentence_embedding = torch.mean(embeddings, dim=1)
     
         print(sentence_embedding)
-        all_embeddings.append(sentence_embedding)
+        all_embeddings.append(sentence_embedding.tolist())
     return all_embeddings