akhaliq HF Staff commited on
Commit
bdb1d57
·
1 Parent(s): 03ae259

Update Gradio wheel version and add show_nemotron_prompt node

Browse files
Files changed (2) hide show
  1. app.py +10 -2
  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", "generate_z_image.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