wt002 commited on
Commit
bfaea4a
·
verified ·
1 Parent(s): 46ff440

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -271,7 +271,7 @@ class BasicAgent:
271
  # Initialize tokenizer
272
  self.model_id = "codellama/CodeLlama-7b-Instruct-hf"
273
  self.tokenizer = AutoTokenizer.from_pretrained(self.model_id, token=token)
274
-
275
  # Model (e.g., HfApiModel or other)
276
  self.model = HfApiModel(
277
  model=self.model_id,
@@ -299,7 +299,7 @@ Answer should be a short string, number, or comma-separated list. Keep it brief.
299
  """
300
 
301
  self.agent = CodeAgent(
302
- model=model,
303
  tools=[
304
  search_tool, wiki_search_tool, str_reverse_tool,
305
  keywords_extract_tool, speech_to_text_tool,
 
271
  # Initialize tokenizer
272
  self.model_id = "codellama/CodeLlama-7b-Instruct-hf"
273
  self.tokenizer = AutoTokenizer.from_pretrained(self.model_id, token=token)
274
+
275
  # Model (e.g., HfApiModel or other)
276
  self.model = HfApiModel(
277
  model=self.model_id,
 
299
  """
300
 
301
  self.agent = CodeAgent(
302
+ model=self.model,
303
  tools=[
304
  search_tool, wiki_search_tool, str_reverse_tool,
305
  keywords_extract_tool, speech_to_text_tool,