Update app.py
Browse files
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 |
-
|
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(
|