luke9705 commited on
Commit
7b2bd1f
·
1 Parent(s): e06b8fa

Update HfApiModel to use the 'google/gemma-3-27b-it' model in the Agent class.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -156,7 +156,7 @@ def generate_audio(prompt: str) -> object:
156
  ## agent definition
157
  class Agent:
158
  def __init__(self, ):
159
- client = HfApiModel("deepseek-ai/DeepSeek-R1-0528", provider="nebius", api_key=os.getenv("NEBIUS_API_KEY"))
160
  self.agent = CodeAgent(
161
  model=client,
162
  tools=[DuckDuckGoSearchTool(max_results=5), VisitWebpageTool(max_output_length=20000), generate_image, download_images, transcribe_audio],
 
156
  ## agent definition
157
  class Agent:
158
  def __init__(self, ):
159
+ client = HfApiModel("google/gemma-3-27b-it", provider="nebius", api_key=os.getenv("NEBIUS_API_KEY"))
160
  self.agent = CodeAgent(
161
  model=client,
162
  tools=[DuckDuckGoSearchTool(max_results=5), VisitWebpageTool(max_output_length=20000), generate_image, download_images, transcribe_audio],