name chroma

src/visual_qa.py

@@ -26,20 +26,7 @@ def initialize_collection(collection_name="clip_image_embeddings"):
 
     return collection
 
-def initialize_collection(collection_name="clip_image_embeddings"):
-    client = chromadb.PersistentClient(path="./chroma_db")
-
-    # Extract just the collection names
-    existing_collections = [col.name for col in client.list_collections()]
 
-    if collection_name in existing_collections:
-        collection = client.get_collection(name=collection_name)
-        print(f"Using existing collection: {collection_name}")
-    else:
-        collection = client.create_collection(name=collection_name)
-        print(f"Created new collection: {collection_name}")
-
-    return collection
 
 def main_create_image_collection(df_emb, collection_name="clip_image_embeddings"):
     documents = []