yoda1976 commited on
Commit
3e76bdc
·
1 Parent(s): 4b62555

Add repo id instead of endpint

Browse files
Files changed (1) hide show
  1. tools.py +1 -1
tools.py CHANGED
@@ -47,7 +47,7 @@ def web_search(query: str) -> str:
47
  return context if context else "No results found."
48
 
49
  llm = ChatHuggingFace(llm = HuggingFaceEndpoint(
50
- endpoint_url="https://api-inference.huggingface.co/models/meta-llama/Llama-2-7b-chat-hf",
51
  temperature=0,
52
  huggingfacehub_api_token=os.environ.get("HUGGING_FACE_API_KEY")))
53
  tools = [add, subtract, divide, web_search]
 
47
  return context if context else "No results found."
48
 
49
  llm = ChatHuggingFace(llm = HuggingFaceEndpoint(
50
+ repo_id = "meta-llama/Llama-2-7b-chat-hf",
51
  temperature=0,
52
  huggingfacehub_api_token=os.environ.get("HUGGING_FACE_API_KEY")))
53
  tools = [add, subtract, divide, web_search]