DeepVen commited on
Commit
9ae0443
1 Parent(s): 9f872b3

Upload Venkat.py

Browse files
Files changed (1) hide show
  1. Venkat.py +2 -2
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(embeddings)
201
  # llm = HuggingFaceHub(
202
  # repo_id="google/flan-t5-xxl",
203
  # model_kwargs={"temperature": 1, "max_length": 1000000},
@@ -213,6 +213,6 @@ load embedding and models for extractor during start up
213
  '''
214
 
215
  # Create extractor instance
216
- extractor = Extractor(path="google/flan-t5-base")
217
  #extractor = Extractor(embeddings, "TheBloke/Llama-2-7B-GGUF")
218
  #extractor = Extractor(embeddings, "google/flan-t5-xl")
 
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},
 
213
  '''
214
 
215
  # Create extractor instance
216
+ #extractor = Extractor(path="google/flan-t5-large")
217
  #extractor = Extractor(embeddings, "TheBloke/Llama-2-7B-GGUF")
218
  #extractor = Extractor(embeddings, "google/flan-t5-xl")