jarguello76 commited on
Commit
de5567b
·
verified ·
1 Parent(s): 1006b91

Update tools.py

Browse files
Files changed (1) hide show
  1. tools.py +6 -5
tools.py CHANGED
@@ -139,12 +139,13 @@ def build_graph(csv_file_path: str = "embeddings.csv"):
139
  from langchain_community.chat_models import ChatHuggingFace
140
  from huggingface_hub import InferenceClient
141
 
142
- client = InferenceClient(
143
- model="meta-llama/Meta-Llama-3-8B-Instruct",
144
- token=os.environ["agents_token"]
 
 
145
  )
146
-
147
- llm = ChatHuggingFace(client=client)
148
 
149
 
150
 
 
139
  from langchain_community.chat_models import ChatHuggingFace
140
  from huggingface_hub import InferenceClient
141
 
142
+ from langchain_community.chat_models import ChatHuggingFace
143
+
144
+ llm = ChatHuggingFace(
145
+ repo_id="meta-llama/Meta-Llama-3-8B-Instruct",
146
+ huggingfacehub_api_token=os.environ["agents_token"]
147
  )
148
+
 
149
 
150
 
151