Update tooling.py
Browse files- tooling.py +1 -1
tooling.py
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
from smolagents import Tool
|
| 2 |
import hashlib
|
| 3 |
import json
|
| 4 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
|
|
| 1 |
+
from smolagents import Tool, HfApiModel
|
| 2 |
import hashlib
|
| 3 |
import json
|
| 4 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|