Update Index.py

Index.py

@@ -226,7 +226,8 @@ def rag(domain: str, question: str):
     # if db_exists:
     embeddings = _load_embeddings_from_db(db_exists, domain)
     # Create extractor instance
-    extractor = Extractor(embeddings, "google/flan-t5-base")
+    # extractor = Extractor(embeddings, "google/flan-t5-base")
+    extractor = Extractor(embeddings, "TheBloke/Llama-2-7B-GGUF/llama-2-7b.Q4_0.gguf")
     # llm = HuggingFaceHub(
     #     repo_id="google/flan-t5-xxl",
     #     model_kwargs={"temperature": 1, "max_length": 1000000},