linoyts HF Staff commited on
Commit
73cbe25
·
verified ·
1 Parent(s): 34ee4d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +41 -46
app.py CHANGED
@@ -90,77 +90,72 @@ with gr.Blocks() as demo:
90
  placeholder="Enter your image description...",
91
  lines=4,
92
  )
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
93
 
94
- with gr.Row():
95
- height = gr.Slider(
96
- minimum=512,
97
- maximum=2048,
98
- value=1024,
99
- step=64,
100
- label="Height",
101
- )
102
- width = gr.Slider(
103
- minimum=512,
104
- maximum=2048,
105
- value=1024,
106
- step=64,
107
- label="Width",
108
- )
109
-
110
- with gr.Row():
111
- num_inference_steps = gr.Slider(
112
- minimum=1,
113
- maximum=20,
114
- value=9,
115
- step=1,
116
- label="Inference Steps",
117
- info="9 steps results in 8 DiT forwards",
118
- )
119
-
120
- with gr.Row():
121
- seed = gr.Number(
122
- label="Seed",
123
- value=42,
124
- precision=0,
125
- )
126
- randomize_seed = gr.Checkbox(
127
- label="Randomize Seed",
128
- value=True,
129
- )
130
-
131
- generate_btn = gr.Button("🚀 Generate", variant="primary", size="lg")
132
 
133
  with gr.Column(scale=1):
134
  output_image = gr.ImageSlider(
135
  label="Generated Image",
136
  type="pil",
137
  )
138
- used_seed = gr.Number(
139
- label="Seed Used",
140
- interactive=False,
141
- )
142
 
143
- gr.Markdown("### 💡 Example Prompts")
144
  gr.Examples(
145
  examples=examples,
146
  inputs=[prompt],
147
  cache_examples=False,
148
  )
149
 
150
- gr.Markdown("Demo by [mrfakename](https://x.com/realmrfakename). Model by Alibaba. The model is licensed under Apache 2.0, you can use generated images commercially! Thanks to [multimodalart](https://huggingface.co/multimodalart) for the FA3 + AoTI enhancements/speedups")
151
 
152
  # Connect the generate button
153
  generate_btn.click(
154
  fn=generate_image,
155
  inputs=[prompt, height, width, num_inference_steps, seed, randomize_seed],
156
- outputs=[output_image, used_seed],
157
  )
158
 
159
  # Also allow generating by pressing Enter in the prompt box
160
  prompt.submit(
161
  fn=generate_image,
162
  inputs=[prompt, height, width, num_inference_steps, seed, randomize_seed],
163
- outputs=[output_image, used_seed],
164
  )
165
 
166
  if __name__ == "__main__":
 
90
  placeholder="Enter your image description...",
91
  lines=4,
92
  )
93
+ with gr.Accordion("Advanced Settings", open=False):
94
+ with gr.Row():
95
+ height = gr.Slider(
96
+ minimum=512,
97
+ maximum=2048,
98
+ value=1024,
99
+ step=64,
100
+ label="Height",
101
+ )
102
+ width = gr.Slider(
103
+ minimum=512,
104
+ maximum=2048,
105
+ value=1024,
106
+ step=64,
107
+ label="Width",
108
+ )
109
+
110
+ with gr.Row():
111
+ num_inference_steps = gr.Slider(
112
+ minimum=1,
113
+ maximum=20,
114
+ value=9,
115
+ step=1,
116
+ label="Inference Steps",
117
+ info="9 steps results in 8 DiT forwards",
118
+ )
119
+
120
+ with gr.Row():
121
+ seed = gr.Number(
122
+ label="Seed",
123
+ value=42,
124
+ precision=0,
125
+ )
126
+ randomize_seed = gr.Checkbox(
127
+ label="Randomize Seed",
128
+ value=True,
129
+ )
130
 
131
+ generate_btn = gr.Button("Generate", variant="primary", size="lg")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
132
 
133
  with gr.Column(scale=1):
134
  output_image = gr.ImageSlider(
135
  label="Generated Image",
136
  type="pil",
137
  )
138
+
 
 
 
139
 
 
140
  gr.Examples(
141
  examples=examples,
142
  inputs=[prompt],
143
  cache_examples=False,
144
  )
145
 
 
146
 
147
  # Connect the generate button
148
  generate_btn.click(
149
  fn=generate_image,
150
  inputs=[prompt, height, width, num_inference_steps, seed, randomize_seed],
151
+ outputs=[output_image, seed],
152
  )
153
 
154
  # Also allow generating by pressing Enter in the prompt box
155
  prompt.submit(
156
  fn=generate_image,
157
  inputs=[prompt, height, width, num_inference_steps, seed, randomize_seed],
158
+ outputs=[output_image, seed],
159
  )
160
 
161
  if __name__ == "__main__":