Spaces:
Sleeping
Sleeping
jesusgj
commited on
Commit
·
914cd84
1
Parent(s):
5b4c1c3
Modified files
Browse files
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
|
| 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 =
|
| 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
|