DeepVen commited on
Commit
75e2955
1 Parent(s): 75d63f6

Upload Venkat.py

Browse files
Files changed (1) hide show
  1. Venkat.py +1 -1
Venkat.py CHANGED
@@ -197,7 +197,7 @@ def rag(domain: str, question: str):
197
  print(db_exists)
198
  # if db_exists:
199
  embeddings = _load_embeddings_from_db(db_exists, domain)
200
- extractor = Extractor(similarity=embeddings, path="google/flan-t5-large")
201
  # llm = HuggingFaceHub(
202
  # repo_id="google/flan-t5-xxl",
203
  # model_kwargs={"temperature": 1, "max_length": 1000000},
 
197
  print(db_exists)
198
  # if db_exists:
199
  embeddings = _load_embeddings_from_db(db_exists, domain)
200
+ extractor = Extractor(similarity=embeddings, path="google/flan-t5-base")
201
  # llm = HuggingFaceHub(
202
  # repo_id="google/flan-t5-xxl",
203
  # model_kwargs={"temperature": 1, "max_length": 1000000},