vechism commited on
Commit
a8ed50b
·
verified ·
1 Parent(s): 63240c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -11,10 +11,10 @@ from Gradio_UI import GradioUI
11
  @tool
12
  def mutant_unicorn(animal_name:str) -> str:
13
  """
14
- A tool that generates an image of a magic mutant unicorn give an animal name.
15
 
16
  Args:
17
- animal_name: A string containing an animal name, real or mytical. (e.g., 'lion, dragon, worm, etc')
18
 
19
  Returns:
20
  str: The generated image or an error message
@@ -60,7 +60,7 @@ custom_role_conversions=None,
60
 
61
 
62
  # Import tool from Hub
63
- image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
64
 
65
  with open("prompts.yaml", 'r') as stream:
66
  prompt_templates = yaml.safe_load(stream)
 
11
  @tool
12
  def mutant_unicorn(animal_name:str) -> str:
13
  """
14
+ A tool that generates an image of a mutant unicorn given an animal name.
15
 
16
  Args:
17
+ animal_name: A string containing the name of an animal, real or mytical. (e.g., 'lion, dragon, worm, etc')
18
 
19
  Returns:
20
  str: The generated image or an error message
 
60
 
61
 
62
  # Import tool from Hub
63
+ # image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
64
 
65
  with open("prompts.yaml", 'r') as stream:
66
  prompt_templates = yaml.safe_load(stream)