Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -64,7 +64,7 @@ model = HfApiModel(
|
|
64 |
)
|
65 |
|
66 |
# Import tool from Hub (example: text-to-image tool)
|
67 |
-
image_generation_tool = load_tool("
|
68 |
|
69 |
with open("prompts.yaml", 'r') as stream:
|
70 |
prompt_templates = yaml.safe_load(stream)
|
@@ -72,10 +72,7 @@ with open("prompts.yaml", 'r') as stream:
|
|
72 |
# Update the agent's tools list to include multiple tools:
|
73 |
agent = CodeAgent(
|
74 |
model=model,
|
75 |
-
tools=[
|
76 |
-
final_answer,
|
77 |
-
duck_search_tool,
|
78 |
-
get_latest_ai_news],
|
79 |
max_steps=6,
|
80 |
verbosity_level=1,
|
81 |
grammar=None,
|
|
|
64 |
)
|
65 |
|
66 |
# Import tool from Hub (example: text-to-image tool)
|
67 |
+
image_generation_tool = load_tool("MohammedNasser/text-to-image", trust_remote_code=True)
|
68 |
|
69 |
with open("prompts.yaml", 'r') as stream:
|
70 |
prompt_templates = yaml.safe_load(stream)
|
|
|
72 |
# Update the agent's tools list to include multiple tools:
|
73 |
agent = CodeAgent(
|
74 |
model=model,
|
75 |
+
tools=[final_answer],
|
|
|
|
|
|
|
76 |
max_steps=6,
|
77 |
verbosity_level=1,
|
78 |
grammar=None,
|