TheBobBob commited on
Commit
fe1d03d
·
verified ·
1 Parent(s): 4801d61

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -10
app.py CHANGED
@@ -139,16 +139,8 @@ def create_vector_db(final_items):
139
  global db
140
  client = chromadb.Client()
141
  collection_name = "BioModelsRAG"
142
-
143
- try:
144
- db = client.create_collection(
145
- name=collection_name,
146
- metadata={"hnsw:space": "cosine"},
147
- embedding_function="all-MiniLM-L6-v2" # Corrected spelling
148
- )
149
- except UniqueConstraintError:
150
- print(f"Collection '{collection_name}' already exists.")
151
- db = client.get_collection(name=collection_name, embedding_function="all-MiniLM-L6-v2") # Adjusted to use `client`
152
 
153
  # Initialize Llama model
154
  llm = Llama.from_pretrained(
 
139
  global db
140
  client = chromadb.Client()
141
  collection_name = "BioModelsRAG"
142
+ print(f"Collection '{collection_name}' already exists.")
143
+ db = client.get_collection(name=collection_name, embedding_function="all-MiniLM-L6-v2")
 
 
 
 
 
 
 
 
144
 
145
  # Initialize Llama model
146
  llm = Llama.from_pretrained(