Update app.py
Browse files
app.py
CHANGED
@@ -5,8 +5,8 @@ import inspect
|
|
5 |
import pandas as pd
|
6 |
import torch, spaces
|
7 |
from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
|
8 |
-
import
|
9 |
-
from smolagents import
|
10 |
from huggingface_hub import InferenceClient, hf_hub_download
|
11 |
|
12 |
# (Keep Constants as is)
|
@@ -50,7 +50,6 @@ class BasicAgent:
|
|
50 |
def __init__(self):
|
51 |
print("BasicAgent initialized.")
|
52 |
print("ENV-HF_TOKEN-LEN", len(os.environ["HF_TOKEN"]), file=sys.stderr)
|
53 |
-
print(smolagents.__version__)
|
54 |
check_token_access()
|
55 |
|
56 |
# Local test
|
|
|
5 |
import pandas as pd
|
6 |
import torch, spaces
|
7 |
from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
|
8 |
+
from smolagents import CodeAgent, HfApiModel
|
9 |
+
from smolagents.llm import FunctionLLM
|
10 |
from huggingface_hub import InferenceClient, hf_hub_download
|
11 |
|
12 |
# (Keep Constants as is)
|
|
|
50 |
def __init__(self):
|
51 |
print("BasicAgent initialized.")
|
52 |
print("ENV-HF_TOKEN-LEN", len(os.environ["HF_TOKEN"]), file=sys.stderr)
|
|
|
53 |
check_token_access()
|
54 |
|
55 |
# Local test
|