Update Gradio wheel version and add show_nemotron_prompt node
Browse files- app.py +10 -2
- requirements.txt +2 -3
app.py
CHANGED
|
@@ -75,6 +75,13 @@ def generate_prompt(concept: str) -> str:
|
|
| 75 |
return f"A detailed, high-quality, professional commercial product photograph of {concept}"
|
| 76 |
|
| 77 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 78 |
def generate_z_image(prompt: str) -> dict:
|
| 79 |
"""
|
| 80 |
Generates an image from a prompt using the Tongyi-MAI/Z-Image-Turbo model.
|
|
@@ -205,9 +212,10 @@ class LockedWorkflow(gr.Workflow):
|
|
| 205 |
|
| 206 |
|
| 207 |
demo = LockedWorkflow(
|
| 208 |
-
bind=[generate_prompt, generate_z_image],
|
| 209 |
edges=[
|
| 210 |
-
("generate_prompt", "
|
|
|
|
| 211 |
]
|
| 212 |
)
|
| 213 |
|
|
|
|
| 75 |
return f"A detailed, high-quality, professional commercial product photograph of {concept}"
|
| 76 |
|
| 77 |
|
| 78 |
+
def show_nemotron_prompt(prompt: str) -> str:
|
| 79 |
+
"""
|
| 80 |
+
Displays the detailed image prompt generated by the NVIDIA Nemotron model.
|
| 81 |
+
"""
|
| 82 |
+
return prompt
|
| 83 |
+
|
| 84 |
+
|
| 85 |
def generate_z_image(prompt: str) -> dict:
|
| 86 |
"""
|
| 87 |
Generates an image from a prompt using the Tongyi-MAI/Z-Image-Turbo model.
|
|
|
|
| 212 |
|
| 213 |
|
| 214 |
demo = LockedWorkflow(
|
| 215 |
+
bind=[generate_prompt, show_nemotron_prompt, generate_z_image],
|
| 216 |
edges=[
|
| 217 |
+
("generate_prompt", "show_nemotron_prompt.prompt"),
|
| 218 |
+
("show_nemotron_prompt", "generate_z_image.prompt"),
|
| 219 |
]
|
| 220 |
)
|
| 221 |
|
requirements.txt
CHANGED
|
@@ -1,5 +1,4 @@
|
|
| 1 |
-
gradio
|
| 2 |
git+https://github.com/huggingface/diffusers
|
| 3 |
transformers
|
| 4 |
-
kernels
|
| 5 |
-
gradio[mcp]
|
|
|
|
| 1 |
+
gradio[oauth] @ https://huggingface.co/buckets/gradio/pypi-previews/resolve/eb2cebf05ac60e14832bcee09894832ab0137c9f/gradio-6.17.3-py3-none-any.whl
|
| 2 |
git+https://github.com/huggingface/diffusers
|
| 3 |
transformers
|
| 4 |
+
kernels
|
|
|