akshatsoni commited on
Commit
5a60cb7
·
verified ·
1 Parent(s): f4b847b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -49,7 +49,8 @@ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_co
49
 
50
  # Load system prompt from prompt.yaml file
51
  with open("prompts.yaml", 'r') as stream:
52
- prompt_templates = yaml.safe_load(stream)
 
53
 
54
  print("prompt_templates ====>", prompt_templates )
55
 
 
49
 
50
  # Load system prompt from prompt.yaml file
51
  with open("prompts.yaml", 'r') as stream:
52
+ prompt_templates = yaml.safe_load(stream)["prompt_templates"]
53
+
54
 
55
  print("prompt_templates ====>", prompt_templates )
56