nirmalaag commited on
Commit
ccfc791
·
verified ·
1 Parent(s): a8dab95

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -65,7 +65,7 @@ def main():
65
  #PATH = 'model/'
66
  #llm = AutoModelForCausalLM.from_pretrained("CohereForAI/aya-101")
67
  # llm = AutoModelForCausalLM.from_pretrained(PATH,local_files_only=True)
68
- llm = huggingface_hub.HuggingFaceHub(repo_id="google/flan-ul2",
69
  model_kwargs={"temperature":1.0, "max_length":100})
70
  docs = vector_store.similarity_search(query=query, k=1)
71
  global chain
 
65
  #PATH = 'model/'
66
  #llm = AutoModelForCausalLM.from_pretrained("CohereForAI/aya-101")
67
  # llm = AutoModelForCausalLM.from_pretrained(PATH,local_files_only=True)
68
+ llm = huggingface_hub.HuggingFaceHub(repo_id="google/flan-t5-base",
69
  model_kwargs={"temperature":1.0, "max_length":100})
70
  docs = vector_store.similarity_search(query=query, k=1)
71
  global chain