supratipb commited on
Commit
949b064
·
verified ·
1 Parent(s): 1abac52

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ class BasicAgent:
18
  grok_model = pipeline("text-generation", model="distilgpt2")
19
  openAI_model = OpenAIServerModel(model_id="gpt-4o-mini", api_key="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")
20
  self.agent = CodeAgent(
21
- model = openAI_model,
22
  tools = [DuckDuckGoSearchTool(), WikipediaSearchTool()],
23
  add_base_tools = True,
24
  )
 
18
  grok_model = pipeline("text-generation", model="distilgpt2")
19
  openAI_model = OpenAIServerModel(model_id="gpt-4o-mini", api_key="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")
20
  self.agent = CodeAgent(
21
+ model = grok_model,
22
  tools = [DuckDuckGoSearchTool(), WikipediaSearchTool()],
23
  add_base_tools = True,
24
  )