Changed LLM to DeepSeek R1
Browse files
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("
|
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],
|