Spaces:
Running
Running
Update app.py
Browse files
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
|
| 15 |
|
| 16 |
Args:
|
| 17 |
-
animal_name: A string containing an animal
|
| 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)
|