vsanchezn commited on
Commit
c793910
·
verified ·
1 Parent(s): 52720d0

changing model

Browse files
Files changed (1) hide show
  1. app.py +9 -7
app.py CHANGED
@@ -91,13 +91,15 @@ def get_current_time_in_timezone(timezone: str) -> str:
91
 
92
 
93
  final_answer = FinalAnswerTool()
94
- model = HfApiModel(
95
- max_tokens=2096,
96
- temperature=0.5,
97
- model_id='https://wxknx1kg971u7k1n.us-east-1.aws.endpoints.huggingface.cloud',# it is possible that this model may be overloaded
98
- custom_role_conversions=None,
99
- )
100
-
 
 
101
 
102
  # # Import tool from Hub
103
  # image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
91
 
92
 
93
  final_answer = FinalAnswerTool()
94
+ # model = HfApiModel(
95
+ # max_tokens=2096,
96
+ # temperature=0.5,
97
+ # model_id='https://wxknx1kg971u7k1n.us-east-1.aws.endpoints.huggingface.cloud',# it is possible that this model may be overloaded
98
+ # custom_role_conversions=None,
99
+ # )
100
+ model = LiteLLMModel(
101
+ model_id="ollama/qwen2.5-coder:32b"
102
+ )
103
 
104
  # # Import tool from Hub
105
  # image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)