ambadkar commited on
Commit
86aed95
·
verified ·
1 Parent(s): 6747959

trying to fix image generation

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