Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -104,10 +104,13 @@ model_id='sarvamai/sarvam-m',# it is possible that this model may be overloaded
|
|
104 |
)
|
105 |
|
106 |
|
|
|
107 |
# Import tool from Hub
|
108 |
image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
|
109 |
|
110 |
-
|
|
|
|
|
111 |
|
112 |
agent = CodeAgent(
|
113 |
model=model,
|
@@ -118,6 +121,7 @@ agent = CodeAgent(
|
|
118 |
planning_interval=None,
|
119 |
name=None,
|
120 |
description=None,
|
|
|
121 |
|
122 |
)
|
123 |
|
|
|
104 |
)
|
105 |
|
106 |
|
107 |
+
|
108 |
# Import tool from Hub
|
109 |
image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
|
110 |
|
111 |
+
with open("prompts.yaml", 'r') as stream:
|
112 |
+
prompt_templates = yaml.safe_load(stream)
|
113 |
+
|
114 |
|
115 |
agent = CodeAgent(
|
116 |
model=model,
|
|
|
121 |
planning_interval=None,
|
122 |
name=None,
|
123 |
description=None,
|
124 |
+
prompt_templates=prompt_templates
|
125 |
|
126 |
)
|
127 |
|