krrishk22 commited on
Commit
9660875
·
verified ·
1 Parent(s): f5ab761

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
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