Update app.py
Browse files
app.py
CHANGED
@@ -149,7 +149,7 @@ def create_vector_db(final_items):
|
|
149 |
)
|
150 |
except UniqueConstraintError:
|
151 |
print(f"Collection '{collection_name}' already exists.")
|
152 |
-
db = client.get_collection(name=collection_name) # Adjusted to use `client`
|
153 |
|
154 |
# Initialize Llama model
|
155 |
llm = Llama.from_pretrained(
|
|
|
149 |
)
|
150 |
except UniqueConstraintError:
|
151 |
print(f"Collection '{collection_name}' already exists.")
|
152 |
+
db = client.get_collection(name=collection_name, embedding_function="all-MiniLM-L6-v2") # Adjusted to use `client`
|
153 |
|
154 |
# Initialize Llama model
|
155 |
llm = Llama.from_pretrained(
|