luke9705 commited on
Commit
56f7f57
·
1 Parent(s): 59a50d1

Update agent initialization: switch to OpenAIServerModel and update model ID

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -198,12 +198,12 @@ def generate_audio_from_sample(prompt: str, duration: int, sample: list[int, np.
198
  ## agent definition
199
  class Agent:
200
  def __init__(self, ):
201
- client = HfApiModel("deepseek-ai/DeepSeek-R1-0528", provider="nebius", api_key=os.getenv("NEBIUS_API_KEY"))
202
- """client = OpenAIServerModel(
203
- model_id="claude-sonnet-4-20250514",
204
  api_base="https://api.anthropic.com/v1/",
205
  api_key=os.environ["ANTHROPIC_API_KEY"],
206
- )"""
207
  self.agent = CodeAgent(
208
  model=client,
209
  tools=[DuckDuckGoSearchTool(max_results=5),
 
198
  ## agent definition
199
  class Agent:
200
  def __init__(self, ):
201
+ #client = HfApiModel("deepseek-ai/DeepSeek-R1-0528", provider="nebius", api_key=os.getenv("NEBIUS_API_KEY"))
202
+ client = OpenAIServerModel(
203
+ model_id="claude-opus-4-20250514",
204
  api_base="https://api.anthropic.com/v1/",
205
  api_key=os.environ["ANTHROPIC_API_KEY"],
206
+ )
207
  self.agent = CodeAgent(
208
  model=client,
209
  tools=[DuckDuckGoSearchTool(max_results=5),