jesusgj commited on
Commit
914cd84
·
1 Parent(s): 5b4c1c3

Modified files

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -4,7 +4,7 @@ import os
4
  import re
5
  import requests
6
  from smolagents import CodeAgent, ToolCallingAgent, WebSearchTool, tool
7
- from smolagents.models.huggingface import HuggingFaceModel
8
  from dotenv import load_dotenv
9
  from markdownify import markdownify
10
  from requests.exceptions import RequestException
@@ -17,7 +17,7 @@ def initialize_agent():
17
  # Make sure to set HF_TOKEN in your Hugging Face Space secrets
18
  model_name = "HuggingFaceH4/zephyr-7b-beta"
19
  try:
20
- model = HuggingFaceModel(model=model_name, token=os.environ.get("HF_TOKEN"))
21
  except Exception as e:
22
  print(f"Error loading model: {e}")
23
  model = None
 
4
  import re
5
  import requests
6
  from smolagents import CodeAgent, ToolCallingAgent, WebSearchTool, tool
7
+ from smolagents import InferenceClientModel
8
  from dotenv import load_dotenv
9
  from markdownify import markdownify
10
  from requests.exceptions import RequestException
 
17
  # Make sure to set HF_TOKEN in your Hugging Face Space secrets
18
  model_name = "HuggingFaceH4/zephyr-7b-beta"
19
  try:
20
+ model = InferenceClientModel(model_id=model_name, token=os.environ.get("HF_TOKEN"))
21
  except Exception as e:
22
  print(f"Error loading model: {e}")
23
  model = None