Upload Index.py

Index.py

@@ -239,7 +239,8 @@ def rag(domain: str, question: str):
     #extractor = Extractor(embeddings, "google/flan-t5-base")
     #extractor = Extractor(embeddings, "TheBloke/Llama-2-7B-GGUF")
     print("before calling extractor")
-    extractor = Extractor(embeddings, "distilbert-base-cased-distilled-squad")
+    #extractor = Extractor(embeddings, "distilbert-base-cased-distilled-squad")
+    extractor = Extractor(embeddings, "google/flan-t5-base")
     # llm = HuggingFaceHub(
     #     repo_id="google/flan-t5-xxl",
     #     model_kwargs={"temperature": 1, "max_length": 1000000},