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
|