berkod-ai commited on
Commit
dbbd7fd
·
verified ·
1 Parent(s): 97d48d7

Update app.py

Browse files

image generation tool has been modified

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -48,7 +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:
54
  prompt_templates = yaml.safe_load(stream)
 
48
 
49
 
50
  # Import tool from Hub
51
+ image_generation_tool = load_tool("black-forest-labs/FLUX.1-dev", trust_remote_code=True)
52
 
53
  with open("prompts.yaml", 'r') as stream:
54
  prompt_templates = yaml.safe_load(stream)