mery22 commited on
Commit
d49f97e
·
verified ·
1 Parent(s): 4983c5f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -64,8 +64,9 @@ bnb_config = BitsAndBytesConfig(
64
  model = AutoModelForCausalLM.from_pretrained(
65
  "mistralai/Mistral-7B-Instruct-v0.1",quantization_config=bnb_config,
66
  )
 
67
  # Connect query to FAISS index using a retriever
68
- retriever = db.as_retriever(
69
  search_type="mmr",
70
  search_kwargs={'k': 1}
71
  )
 
64
  model = AutoModelForCausalLM.from_pretrained(
65
  "mistralai/Mistral-7B-Instruct-v0.1",quantization_config=bnb_config,
66
  )
67
+ new_db = FAISS.load_local("faiss_index", embeddings)
68
  # Connect query to FAISS index using a retriever
69
+ retriever = new_db.as_retriever(
70
  search_type="mmr",
71
  search_kwargs={'k': 1}
72
  )