antonioschiro commited on
Commit
766e143
·
verified ·
1 Parent(s): 6ec80af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -42,11 +42,12 @@ search_tool = DuckDuckGoSearchRun()
42
  tools = [search_tool]
43
 
44
  # LLM
45
- model = HuggingFaceEndpoint(
46
- repo_id="mistralai/Mistral-7B-Instruct-v0.3",
47
- temperature = 0.1,
 
 
48
  )
49
-
50
  class BasicAgent:
51
  def __init__(self):
52
  self.agent = create_react_agent(model, tools, prompt)
 
42
  tools = [search_tool]
43
 
44
  # LLM
45
+ model = ChatHuggingFace(
46
+ llm=HuggingFaceEndpoint(
47
+ url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
48
+ temperature=0,
49
+ ),
50
  )
 
51
  class BasicAgent:
52
  def __init__(self):
53
  self.agent = create_react_agent(model, tools, prompt)