keynes42 commited on
Commit
4d1848f
·
verified ·
1 Parent(s): 996ecf3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -78,8 +78,8 @@ class BasicAgent:
78
 
79
  # Initialize the agent
80
  pipe = load_llm(hf_token = os.environ["HF_TOKEN"])
81
- self.pipe = LocalLLM(pipe)
82
- self.agent = CodeAgent(llm=self.pipe, tools=[], add_base_tools=True)
83
 
84
  # self.agent = CodeAgent(
85
  # model=model,
 
78
 
79
  # Initialize the agent
80
  pipe = load_llm(hf_token = os.environ["HF_TOKEN"])
81
+ self.model = LocalLLM(pipe)
82
+ self.agent = CodeAgent(model=self.model, tools=[], add_base_tools=True)
83
 
84
  # self.agent = CodeAgent(
85
  # model=model,