YoussefSharawy91 commited on
Commit
4d16b4e
·
verified ·
1 Parent(s): 865f371

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -69,7 +69,7 @@ def search_harry_potter_lore(query: str) -> str:
69
  return f"An error occurred: {str(e)}"
70
 
71
  # Initialize the SmolAgent model
72
- model = HfApiModel(model_id="HuggingFaceTB/SmolLM2-1.7B-Instruct", token=hf_token)
73
 
74
  # Create the agent
75
  agent = CodeAgent(tools=[search_harry_potter_lore], model=model)
 
69
  return f"An error occurred: {str(e)}"
70
 
71
  # Initialize the SmolAgent model
72
+ model = HfApiModel(model_id="HuggingFaceTB/SmolLM2-360M", token=hf_token)
73
 
74
  # Create the agent
75
  agent = CodeAgent(tools=[search_harry_potter_lore], model=model)