ankush13r commited on
Commit
b10c5fb
1 Parent(s): b466a33

Update rag.py

Browse files
Files changed (1) hide show
  1. rag.py +1 -2
rag.py CHANGED
@@ -15,10 +15,9 @@ class RAG:
15
  #vectorstore = "vectorestore" # CA only
16
  vectorstore = "index-BAAI_bge-m3-1500-200-recursive_splitter-CA_ES_UE"
17
 
18
- def __init__(self, hf_token, embeddings_model, model_name):
19
 
20
  self.model_name = model_name
21
- self.hf_token = hf_token
22
 
23
  # load vectore store
24
  embeddings = HuggingFaceEmbeddings(model_name=embeddings_model, model_kwargs={'device': 'cpu'})
 
15
  #vectorstore = "vectorestore" # CA only
16
  vectorstore = "index-BAAI_bge-m3-1500-200-recursive_splitter-CA_ES_UE"
17
 
18
+ def __init__(self, embeddings_model):
19
 
20
  self.model_name = model_name
 
21
 
22
  # load vectore store
23
  embeddings = HuggingFaceEmbeddings(model_name=embeddings_model, model_kwargs={'device': 'cpu'})