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(
|