Spaces:
Runtime error
Runtime error
Upload 2 files
Browse files- Index.py +1 -1
- 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.
|
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
|