Update app.py

app.py

@@ -12,7 +12,7 @@ import chromadb
 
 app = FastAPI()
 client = chromadb.PersistentClient(path="/data/chroma_db")
-collection = client.get_or_create_collection(name="knowledge_base", metadata={"hnsw:space": "cosine", "dimension": 512})
+collection = client.get_or_create_collection(name="knowledge_base")
 
 print("Created collection with 512 dimensions!")
 
@@ -24,7 +24,7 @@ collection = client.get_collection(name="knowledge_base")
 print("Collection Embedding Dimension:", collection.metadata)
 
 # Initialize models
-text_model = SentenceTransformer('all-MiniLM-L6-v2')
+text_model = SentenceTransformer('sentence-transformers/all-MiniLM-L12-v2')
 clip_model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32")
 clip_processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")