Bharadwaj-m7 commited on
Commit
f8a2dc1
·
verified ·
1 Parent(s): 8b9e7f7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -22,7 +22,7 @@ import os
22
  from PIL import Image
23
  from smolagents import CodeAgent, GoogleSearchTool, InferenceClientModel, VisitWebpageTool
24
 
25
-
26
 
27
 
28
  @tool
@@ -97,9 +97,7 @@ def create_agent(model_name: str = "llama3.2:3b-instruct-q4_K_M",
97
  print(f"\nCreating agent with model: {model_name}")
98
  print(f"Max steps: {max_steps}")
99
  print(f"Verbosity: {verbosity}\n")
100
- model = InferenceClientModel(
101
- "Qwen/Qwen2.5-Coder-32B-Instruct", provider="together", max_tokens=8096)
102
-
103
  agent = CodeAgent(
104
  model=model,
105
  tools=[
 
22
  from PIL import Image
23
  from smolagents import CodeAgent, GoogleSearchTool, InferenceClientModel, VisitWebpageTool
24
 
25
+ from smolagents import OpenAIServerModel
26
 
27
 
28
  @tool
 
97
  print(f"\nCreating agent with model: {model_name}")
98
  print(f"Max steps: {max_steps}")
99
  print(f"Verbosity: {verbosity}\n")
100
+ model = OpenAIServerModel("gpt-4o", max_tokens=8096)
 
 
101
  agent = CodeAgent(
102
  model=model,
103
  tools=[