themissingCRAM commited on
Commit
e95efe1
·
1 Parent(s): 28fb60e

testing llm again

Browse files
Files changed (1) hide show
  1. app.py +15 -15
app.py CHANGED
@@ -96,18 +96,18 @@ def init_db():
96
 
97
  if __name__ == "__main__":
98
  init_db()
99
- # model = HfApiModel(
100
- # model_id="HuggingFaceH4/zephyr-7b-beta",
101
- # token=os.getenv("my_first_agents_hf_tokens"),
102
- # )
103
-
104
- # agent = CodeAgent(
105
- # tools=[sql_engine_tool],
106
- # # system_prompt="""
107
- # # You are a text to sql converter
108
- # # """,
109
- # model=model,
110
- # max_steps=1,
111
- # verbosity_level=1,
112
- # )
113
- # GradioUI(agent).launch()
 
96
 
97
  if __name__ == "__main__":
98
  init_db()
99
+ model = HfApiModel(
100
+ model_id="HuggingFaceH4/zephyr-7b-beta",
101
+ token=os.getenv("my_first_agents_hf_tokens"),
102
+ )
103
+
104
+ agent = CodeAgent(
105
+ tools=[sql_engine_tool],
106
+ # system_prompt="""
107
+ # You are a text to sql converter
108
+ # """,
109
+ model=model,
110
+ max_steps=1,
111
+ verbosity_level=1,
112
+ )
113
+ GradioUI(agent).launch()