Solobrad commited on
Commit
8a568e9
·
verified ·
1 Parent(s): 20d6ea1

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +5 -2
agent.py CHANGED
@@ -1,5 +1,5 @@
1
 
2
- from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
3
 
4
  from llama_index.embeddings.huggingface import HuggingFaceEmbedding
5
  from langchain_community.document_loaders import WikipediaLoader
@@ -13,7 +13,10 @@ import os
13
  import requests
14
 
15
  hf_token = os.getenv("HF_TOKEN")
16
- llm = HuggingFaceInferenceAPI(repo_id="Qwen/Qwen2.5-VL-32B-Instruct", token=hf_token)
 
 
 
17
 
18
  @tool
19
  def web_search(query: str) -> list:
 
1
 
2
+ from llama_index.llms.huggingface import HuggingFaceInferenceAPI
3
 
4
  from llama_index.embeddings.huggingface import HuggingFaceEmbedding
5
  from langchain_community.document_loaders import WikipediaLoader
 
13
  import requests
14
 
15
  hf_token = os.getenv("HF_TOKEN")
16
+ llm = HuggingFaceInferenceAPI(
17
+ model_name="Qwen/Qwen2.5-VL-32B-Instruct",
18
+ token=os.getenv("HF_TOKEN")
19
+ )
20
 
21
  @tool
22
  def web_search(query: str) -> list: