Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -12,26 +12,31 @@ import yaml
|
|
| 12 |
|
| 13 |
# Import modules and classes from openai and smolagents
|
| 14 |
from smolagents import CodeAgent, HfApiModel, DuckDuckGoSearchTool, VisitWebpageTool
|
| 15 |
-
from tools import FinalAnswerTool
|
| 16 |
-
from huggingface_hub import login
|
| 17 |
|
| 18 |
# Log in to Hugging Face with API_KEY
|
| 19 |
HF_API_KEY = os.getenv("HF_API_KEY")
|
| 20 |
-
login(HF_API_KEY)
|
| 21 |
|
| 22 |
# Set API key for OPENAI
|
| 23 |
# os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
|
| 24 |
|
| 25 |
# Create model
|
| 26 |
# model = OpenAIServerModel(model_id="gpt-4o")
|
| 27 |
-
model = HfApiModel()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 28 |
|
| 29 |
# Create final answer instance
|
| 30 |
-
final_answer = FinalAnswerTool(model)
|
| 31 |
|
| 32 |
# Create prompt template
|
| 33 |
-
with open("prompts.yaml", 'r') as stream:
|
| 34 |
-
prompt_templates = yaml.safe_load(stream)
|
| 35 |
|
| 36 |
# (Keep Constants as is)
|
| 37 |
# --- Constants ---
|
|
@@ -70,7 +75,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
|
|
| 70 |
# 1. Instantiate Agent ( modify this part to create your agent)
|
| 71 |
try:
|
| 72 |
agent = CodeAgent(model=model,
|
| 73 |
-
tools=[DuckDuckGoSearchTool(), VisitWebpageTool()
|
| 74 |
max_steps=20,
|
| 75 |
verbosity_level=2)
|
| 76 |
except Exception as e:
|
|
|
|
| 12 |
|
| 13 |
# Import modules and classes from openai and smolagents
|
| 14 |
from smolagents import CodeAgent, HfApiModel, DuckDuckGoSearchTool, VisitWebpageTool
|
| 15 |
+
# from tools import FinalAnswerTool
|
| 16 |
+
# from huggingface_hub import login
|
| 17 |
|
| 18 |
# Log in to Hugging Face with API_KEY
|
| 19 |
HF_API_KEY = os.getenv("HF_API_KEY")
|
| 20 |
+
# login(HF_API_KEY)
|
| 21 |
|
| 22 |
# Set API key for OPENAI
|
| 23 |
# os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
|
| 24 |
|
| 25 |
# Create model
|
| 26 |
# model = OpenAIServerModel(model_id="gpt-4o")
|
| 27 |
+
# model = HfApiModel()
|
| 28 |
+
|
| 29 |
+
model = InferenceClientModel(
|
| 30 |
+
model_id="Qwen/Qwen3-235B-A22B",
|
| 31 |
+
token=HF_API_KEY
|
| 32 |
+
)
|
| 33 |
|
| 34 |
# Create final answer instance
|
| 35 |
+
# final_answer = FinalAnswerTool(model)
|
| 36 |
|
| 37 |
# Create prompt template
|
| 38 |
+
#with open("prompts.yaml", 'r') as stream:
|
| 39 |
+
# prompt_templates = yaml.safe_load(stream)
|
| 40 |
|
| 41 |
# (Keep Constants as is)
|
| 42 |
# --- Constants ---
|
|
|
|
| 75 |
# 1. Instantiate Agent ( modify this part to create your agent)
|
| 76 |
try:
|
| 77 |
agent = CodeAgent(model=model,
|
| 78 |
+
tools=[DuckDuckGoSearchTool(), VisitWebpageTool()],
|
| 79 |
max_steps=20,
|
| 80 |
verbosity_level=2)
|
| 81 |
except Exception as e:
|