Hugging Face's logo

Spaces:
piyushdev
/
agent_1
Running

App Files Community
piyushdev commited on
Commit
7017e6b
·
verified ·
1 Parent(s): 3675e06

Update app.py

Browse files

Fixed the function for image generation

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -47,7 +47,7 @@ custom_role_conversions=None,
47
 
48
 
49
  # Import tool from Hub
50
- image_generation_tool = tool.from_space(
51
  "black-forest-labs/FLUX.1-schnell",
52
  name="image_generator",
53
  description="Generate an image from a prompt"
 
47
 
48
 
49
  # Import tool from Hub
50
+ image_generation_tool = load_tool(
51
  "black-forest-labs/FLUX.1-schnell",
52
  name="image_generator",
53
  description="Generate an image from a prompt"