Edbin commited on
Commit
4cf4833
·
verified ·
1 Parent(s): 9fc2c7c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -1,9 +1,11 @@
1
- from smolagents import CodeAgent, DuckDuckGoSearchTool, InferenceClientModel, load_tool, tool
2
  import datetime
3
  import requests
4
  import pytz
5
  import yaml
6
  from tools.final_answer import FinalAnswerTool
 
 
7
  @tool
8
  def my_custom_tool(arg1:str, arg2:int)-> str: # it's important to specify the return type
9
  # Keep this format for the tool description / args description but feel free to modify the tool
@@ -29,7 +31,7 @@ def get_current_time_in_timezone(timezone: str) -> str:
29
  except Exception as e:
30
  return f"Error fetching time for timezone '{timezone}': {str(e)}"
31
  final_answer = FinalAnswerTool()
32
- model = InferenceClientModel(
33
  max_tokens=2096,
34
  temperature=0.5,
35
  model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
 
1
+ from smolagents import CodeAgent, DuckDuckGoSearchTool, InferenceTool, load_tool, tool
2
  import datetime
3
  import requests
4
  import pytz
5
  import yaml
6
  from tools.final_answer import FinalAnswerTool
7
+
8
+
9
  @tool
10
  def my_custom_tool(arg1:str, arg2:int)-> str: # it's important to specify the return type
11
  # Keep this format for the tool description / args description but feel free to modify the tool
 
31
  except Exception as e:
32
  return f"Error fetching time for timezone '{timezone}': {str(e)}"
33
  final_answer = FinalAnswerTool()
34
+ model = InferenceTool(
35
  max_tokens=2096,
36
  temperature=0.5,
37
  model_id='Qwen/Qwen2.5-Coder-32B-Instruct',