Tuan197 commited on
Commit
59b68ad
·
verified ·
1 Parent(s): 483bde7

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +3 -3
agent.py CHANGED
@@ -163,9 +163,9 @@ def build_graph(provider: str = "groq"):
163
  # TODO: Add huggingface endpoint
164
  llm = ChatHuggingFace(
165
  llm=HuggingFaceEndpoint(
166
- url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
167
- temperature=0,
168
- ),
169
  )
170
  else:
171
  raise ValueError("Invalid provider. Choose 'google', 'groq' or 'huggingface'.")
 
163
  # TODO: Add huggingface endpoint
164
  llm = ChatHuggingFace(
165
  llm=HuggingFaceEndpoint(
166
+ repo_id="Meta-DeepLearning/llama-2-7b-chat-hf",
167
+ temperature=0
168
+ )
169
  )
170
  else:
171
  raise ValueError("Invalid provider. Choose 'google', 'groq' or 'huggingface'.")