arunprasadh commited on
Commit
01b2e53
·
verified ·
1 Parent(s): bfed22d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -48,6 +48,7 @@ custom_role_conversions=None,
48
 
49
 
50
  # Import tool from Hub
 
51
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
52
 
53
  with open("prompts.yaml", 'r') as stream:
 
48
 
49
 
50
  # Import tool from Hub
51
+ @tool
52
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
53
 
54
  with open("prompts.yaml", 'r') as stream: