NagarajanB1990 commited on
Commit
e69dacb
·
verified ·
1 Parent(s): a4632fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,13 +1,13 @@
1
  import os
2
  from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel
3
 
4
- # Make sure token is set in env
5
- os.environ["HF_TOKEN"] = "hf_xxx..." # for local test only
6
 
7
  def who_is_bot(person_name: str) -> str:
8
  agent = CodeAgent(
9
  tools=[DuckDuckGoSearchTool()],
10
- model=HfApiModel() # no args
11
  )
12
  prompt = (
13
  f"Who is {person_name}? "
 
1
  import os
2
  from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel
3
 
4
+ # Set token for test — OR set in environment (safer)
5
+ hf_token = os.getenv("HF_TOKEN")
6
 
7
  def who_is_bot(person_name: str) -> str:
8
  agent = CodeAgent(
9
  tools=[DuckDuckGoSearchTool()],
10
+ model=HfApiModel(model="HuggingFaceH4/zephyr-7b-beta") # Free public model
11
  )
12
  prompt = (
13
  f"Who is {person_name}? "