luke9705 commited on
Commit
8f19907
·
verified ·
1 Parent(s): b5fc7aa

Update app.py

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