fantos commited on
Commit
b62847f
·
verified ·
1 Parent(s): e9b83f7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -17
app.py CHANGED
@@ -187,20 +187,18 @@ button:hover {
187
 
188
  # Add preloading function for the first example
189
  def preload_example():
190
- # Set the default example as the initial state
191
- return (
192
- default_prompt,
193
- inference(
194
- prompt=default_prompt,
195
- seed=default_settings["seed"],
196
- randomize_seed=default_settings["randomize_seed"],
197
- width=default_settings["width"],
198
- height=default_settings["height"],
199
- guidance_scale=default_settings["guidance_scale"],
200
- num_inference_steps=default_settings["num_inference_steps"],
201
- lora_scale=default_settings["lora_scale"],
202
- )
203
  )
 
204
 
205
  with gr.Blocks(css=custom_css, analytics_enabled=False) as demo:
206
  gr.HTML('<div class="title">Paul Cézanne STUDIO</div>')
@@ -223,6 +221,7 @@ with gr.Blocks(css=custom_css, analytics_enabled=False) as demo:
223
  run_button = gr.Button("Generate", variant="primary", scale=0)
224
 
225
  result = gr.Image(label="Generated Image")
 
226
 
227
  with gr.Accordion("Advanced Settings", open=False, elem_classes="advanced-settings"):
228
  seed = gr.Slider(
@@ -278,7 +277,7 @@ with gr.Blocks(css=custom_css, analytics_enabled=False) as demo:
278
  gr.Examples(
279
  examples=examples,
280
  inputs=[prompt],
281
- outputs=[result, seed],
282
  fn=inference,
283
  cache_examples=True,
284
  )
@@ -297,14 +296,14 @@ with gr.Blocks(css=custom_css, analytics_enabled=False) as demo:
297
  num_inference_steps,
298
  lora_scale,
299
  ],
300
- outputs=[result, seed],
301
  )
302
 
303
- # Preload the first example when the app starts
304
  demo.load(
305
  fn=preload_example,
306
  inputs=None,
307
- outputs=[prompt, [result, seed]],
308
  )
309
 
310
  demo.queue()
 
187
 
188
  # Add preloading function for the first example
189
  def preload_example():
190
+ # Generate the default example image
191
+ image, seed_value = inference(
192
+ prompt=default_prompt,
193
+ seed=default_settings["seed"],
194
+ randomize_seed=default_settings["randomize_seed"],
195
+ width=default_settings["width"],
196
+ height=default_settings["height"],
197
+ guidance_scale=default_settings["guidance_scale"],
198
+ num_inference_steps=default_settings["num_inference_steps"],
199
+ lora_scale=default_settings["lora_scale"],
 
 
 
200
  )
201
+ return default_prompt, image, seed_value
202
 
203
  with gr.Blocks(css=custom_css, analytics_enabled=False) as demo:
204
  gr.HTML('<div class="title">Paul Cézanne STUDIO</div>')
 
221
  run_button = gr.Button("Generate", variant="primary", scale=0)
222
 
223
  result = gr.Image(label="Generated Image")
224
+ seed_output = gr.Number(label="Seed", visible=True)
225
 
226
  with gr.Accordion("Advanced Settings", open=False, elem_classes="advanced-settings"):
227
  seed = gr.Slider(
 
277
  gr.Examples(
278
  examples=examples,
279
  inputs=[prompt],
280
+ outputs=[result, seed_output],
281
  fn=inference,
282
  cache_examples=True,
283
  )
 
296
  num_inference_steps,
297
  lora_scale,
298
  ],
299
+ outputs=[result, seed_output],
300
  )
301
 
302
+ # Preload the first example when the app starts - FIXED
303
  demo.load(
304
  fn=preload_example,
305
  inputs=None,
306
+ outputs=[prompt, result, seed_output],
307
  )
308
 
309
  demo.queue()