themissingCRAM commited on
Commit
592b501
·
1 Parent(s): 9f499f9

agent max steps

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -16,7 +16,10 @@ from sqlalchemy import (
16
  )
17
  import spaces
18
 
 
 
19
  engine = create_engine("sqlite:///:memory:")
 
20
 
21
 
22
  @spaces.GPU
@@ -94,6 +97,7 @@ if __name__ == "__main__":
94
  model_id="HuggingFaceH4/zephyr-7b-beta",
95
  token=os.getenv("my_first_agents_hf_tokens"),
96
  ),
97
- max_steps=3,
 
98
  )
99
  GradioUI(agent).launch()
 
16
  )
17
  import spaces
18
 
19
+ from dotenv import load_dotenv
20
+
21
  engine = create_engine("sqlite:///:memory:")
22
+ load_dotenv()
23
 
24
 
25
  @spaces.GPU
 
97
  model_id="HuggingFaceH4/zephyr-7b-beta",
98
  token=os.getenv("my_first_agents_hf_tokens"),
99
  ),
100
+ max_steps=1,
101
+ verbosity_leve=1,
102
  )
103
  GradioUI(agent).launch()