Update app.py

app.py

@@ -12,7 +12,9 @@ import chromadb
 
 app = FastAPI()
 client = chromadb.PersistentClient(path="/data/chroma_db")
-collection = client.get_or_create_collection(name="knowledge_base")
+collection = client.get_or_create_collection(name="knowledge_base", metadata={"hnsw:space": "cosine", "dimension": 512})
+
+print("Created collection with 512 dimensions!")
 
 pdf_file="Sutures and Suturing techniques.pdf"
 pptx_file="impalnt 1.pptx"