supratipb commited on
Commit
d671d72
Β·
verified Β·
1 Parent(s): 30febcc

Upload agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -296,7 +296,7 @@ def build_graph(provider: str = "huggingface"):
296
  # TODO: Add huggingface endpoint
297
  llm = ChatHuggingFace(
298
  llm=HuggingFaceEndpoint(
299
- url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
300
  temperature=0,
301
  ),
302
  )
 
296
  # TODO: Add huggingface endpoint
297
  llm = ChatHuggingFace(
298
  llm=HuggingFaceEndpoint(
299
+ endpoint_url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
300
  temperature=0,
301
  ),
302
  )