magdap116 commited on
Commit
1ae4f03
·
verified ·
1 Parent(s): 6df8704

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -88,11 +88,12 @@ def load_model(model_name):
88
 
89
  from smolagents import TransformersModel
90
 
91
- model = TransformersModel(
92
- model_id=MODEL_NAME,
93
- max_new_tokens=4096)
94
 
95
 
 
 
 
 
96
 
97
  class BasicAgent:
98
  def __init__(self):
@@ -103,7 +104,7 @@ class BasicAgent:
103
  max_steps=1,
104
  verbosity_level=1,
105
  grammar=None,
106
- planning_interval=3,
107
 
108
  )
109
 
 
88
 
89
  from smolagents import TransformersModel
90
 
 
 
 
91
 
92
 
93
+ model_id = "meta-llama/Llama-3.2-3B-Instruct"
94
+ model = TransformersModel(
95
+ model_id=MODEL_NAME,
96
+ max_new_tokens=512)
97
 
98
  class BasicAgent:
99
  def __init__(self):
 
104
  max_steps=1,
105
  verbosity_level=1,
106
  grammar=None,
107
+ planning_interval=1,
108
 
109
  )
110