thatting commited on
Commit
ecc569c
·
verified ·
1 Parent(s): e1b82dc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -8
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(), final_answer],
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: