luke9705 commited on
Commit
7e77d43
·
verified ·
1 Parent(s): 7e40121

Changed LLM to DeepSeek R1

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