ubermenchh commited on
Commit
3939c81
1 Parent(s): 78fe269

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -62,7 +62,7 @@ def chat():
62
 
63
  vector_db = Chroma.from_documents(
64
  documents,
65
- embeddings = HuggingFaceBgeEmbeddings(model_name=model_name, model_kwargs={'device': 'cuda' if torch.cuda.is_available() else 'cpu'}, encode_kwargs=encode_kwargs)
66
  )
67
  repo_id = "tiiuae/falcon-7b-instruct"
68
  qa_chain = RetrievalQA.from_chain_type(
 
62
 
63
  vector_db = Chroma.from_documents(
64
  documents,
65
+ embedding = HuggingFaceBgeEmbeddings(model_name=model_name, model_kwargs={'device': 'cuda' if torch.cuda.is_available() else 'cpu'}, encode_kwargs=encode_kwargs)
66
  )
67
  repo_id = "tiiuae/falcon-7b-instruct"
68
  qa_chain = RetrievalQA.from_chain_type(