tokeron commited on
Commit
d8b8c5a
1 Parent(s): c7ae8be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -69,6 +69,7 @@ def generate_images(prompt, model, seed, skip):
69
  images = get_images(prompt, skip_layers=skip_layers, model=model, seed=seed)
70
  all_images.append((images[0], f'layer_{max_num_of_layers - skip_layers}'))
71
  yield all_images
 
72
 
73
  with gr.Blocks() as demo:
74
 
 
69
  images = get_images(prompt, skip_layers=skip_layers, model=model, seed=seed)
70
  all_images.append((images[0], f'layer_{max_num_of_layers - skip_layers}'))
71
  yield all_images
72
+ gr.Info('Image generation complete')
73
 
74
  with gr.Blocks() as demo:
75