Ihssane123 commited on
Commit
ecc3b15
·
verified ·
1 Parent(s): 1b6b4a3

Update app.py with smol model

Browse files
Files changed (1) hide show
  1. app.py +4 -7
app.py CHANGED
@@ -1,4 +1,6 @@
1
  from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
 
 
2
  import datetime
3
  import requests
4
  import pytz
@@ -38,14 +40,9 @@ final_answer = FinalAnswerTool()
38
 
39
  # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
40
  #model_id='HuggingFaceTB/SmolLM2-1.7B'
41
- model_id='Qwen/Qwen2.5-Coder-32B-Instruct'
 
42
 
43
- model = HfApiModel(
44
- max_tokens=2096,
45
- temperature=0.5,
46
- model_id=model_id,
47
- custom_role_conversions=None,
48
- )
49
 
50
 
51
  # Import tool from Hub
 
1
  from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
2
+ from transformers import AutoModelForCausalLM, AutoTokenizer
3
+
4
  import datetime
5
  import requests
6
  import pytz
 
40
 
41
  # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
42
  #model_id='HuggingFaceTB/SmolLM2-1.7B'
43
+ # pip install transformers
44
+ model_id = "HuggingFaceTB/SmolLM2-1.7B-Instruct"
45
 
 
 
 
 
 
 
46
 
47
 
48
  # Import tool from Hub