LuisBlanche commited on
Commit
c782169
·
verified ·
1 Parent(s): a99cc27

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,4 +1,5 @@
1
- from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
 
2
  from names_dataset import NameDataset
3
 
4
  import datetime
@@ -53,6 +54,7 @@ temperature=0.5,
53
  custom_role_conversions=None,
54
  )
55
 
 
56
 
57
  # Import tool from Hub
58
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
1
+ from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool,tool, LiteLLMModel
2
+
3
  from names_dataset import NameDataset
4
 
5
  import datetime
 
54
  custom_role_conversions=None,
55
  )
56
 
57
+ model = LiteLLMModel("anthropic/claude-3-5-sonnet-latest", temperature=0.2, max_tokens=10)
58
 
59
  # Import tool from Hub
60
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)