AksharaSachin commited on
Commit
d37ac86
·
verified ·
1 Parent(s): e841310

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -47,7 +47,7 @@ def build_agent():
47
  # )
48
 
49
  llm = HuggingFaceHub(
50
- repo_id="Qwen/Qwen1.5-7B-Chat",
51
  model_kwargs={"temperature": 0.2, "max_new_tokens": 256})
52
 
53
  tools = [use_wikipedia_tool, use_search_tool]
 
47
  # )
48
 
49
  llm = HuggingFaceHub(
50
+ repo_id="google/flan-t5-base",
51
  model_kwargs={"temperature": 0.2, "max_new_tokens": 256})
52
 
53
  tools = [use_wikipedia_tool, use_search_tool]