keynes42 commited on
Commit
8ffabaf
·
verified ·
1 Parent(s): fe6a188

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -18,7 +18,7 @@ class BasicAgent:
18
  print("ENV-HF_TOKEN-LEN", len(os.getenv("HF_TOKEN", "")), file=sys.stderr)
19
 
20
  # Initialize the model
21
- model = HfApiModel("deepseek-ai/DeepSeek-V3-0324")
22
 
23
  # Initialize the tools other than the base tools
24
  # See list of base tools in https://github.com/huggingface/smolagents/blob/main/src/smolagents/default_tools.py
@@ -27,7 +27,6 @@ class BasicAgent:
27
  self.agent = CodeAgent(
28
  model=model,
29
  tools=[],
30
- token=os.environ["HF_TOKEN"],
31
  add_base_tools=True
32
  )
33
 
 
18
  print("ENV-HF_TOKEN-LEN", len(os.getenv("HF_TOKEN", "")), file=sys.stderr)
19
 
20
  # Initialize the model
21
+ model = HfApiModel(model_id="deepseek-ai/DeepSeek-V3-0324", token=os.environ["HF_TOKEN"], max_tokens=2048)
22
 
23
  # Initialize the tools other than the base tools
24
  # See list of base tools in https://github.com/huggingface/smolagents/blob/main/src/smolagents/default_tools.py
 
27
  self.agent = CodeAgent(
28
  model=model,
29
  tools=[],
 
30
  add_base_tools=True
31
  )
32