TheBobBob commited on
Commit
06bc08b
·
verified ·
1 Parent(s): a45612c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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(