Update main.py

main.py

@@ -25,7 +25,7 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-def chunk_text(text, chunk_size=512):
+def chunk_text(text, chunk_size=10):
     return [text[i:i + chunk_size] for i in range(0, len(text), chunk_size)]
 
 @app.post("/get_embeding")
@@ -47,7 +47,7 @@ async def get_embeding(text):
         # Optionally, you can average the token embeddings to get a single vector for the sentence
         sentence_embedding = torch.mean(embeddings, dim=1)
     
-        #print(sentence_embedding)
+        print(sentence_embedding)
         all_embeddings.append(sentence_embedding)
     return all_embeddings