DeepVen commited on
Commit
d95598b
1 Parent(s): 8c3e214

Upload 2 files

Browse files
Files changed (2) hide show
  1. Index.py +1 -1
  2. requirements.txt +2 -1
Index.py CHANGED
@@ -227,7 +227,7 @@ def rag(domain: str, question: str):
227
  embeddings = _load_embeddings_from_db(db_exists, domain)
228
  # Create extractor instance
229
  #extractor = Extractor(embeddings, "google/flan-t5-base")
230
- extractor = Extractor(embeddings, "TheBloke/Llama-2-7B-GGUF/llama-2-7b.Q4_0.gguf")
231
  # llm = HuggingFaceHub(
232
  # repo_id="google/flan-t5-xxl",
233
  # model_kwargs={"temperature": 1, "max_length": 1000000},
 
227
  embeddings = _load_embeddings_from_db(db_exists, domain)
228
  # Create extractor instance
229
  #extractor = Extractor(embeddings, "google/flan-t5-base")
230
+ extractor = Extractor(embeddings, "TheBloke/Llama-2-7B-GGUF/blob/main/llama-2-7b.Q4_K_M.gguf")
231
  # llm = HuggingFaceHub(
232
  # repo_id="google/flan-t5-xxl",
233
  # model_kwargs={"temperature": 1, "max_length": 1000000},
requirements.txt CHANGED
@@ -9,4 +9,5 @@ langchain==0.0.301
9
  langsmith==0.0.40
10
  bs4==0.0.1
11
  pandas==2.1.1
12
- SQLAlchemy==2.0.21
 
 
9
  langsmith==0.0.40
10
  bs4==0.0.1
11
  pandas==2.1.1
12
+ SQLAlchemy==2.0.21
13
+ llama-cpp-python