angelahzyuan commited on
Commit
7701440
·
verified ·
1 Parent(s): 404d36e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -34,7 +34,7 @@ def generate(prompt: str, num_images: int=5, guidance_scale=7.5):
34
 
35
  with gr.Blocks() as demo:
36
  gr.Markdown("# SPIN-Diffusion 1.0 Demo")
37
-
38
  with gr.Row():
39
  prompt_input = gr.Textbox(label="Enter your prompt", placeholder="Type something...", lines=2)
40
  generate_btn = gr.Button("Generate images")
@@ -56,6 +56,7 @@ with gr.Blocks() as demo:
56
  gr.Examples(examples=examples, inputs=prompt_input, fn=generate, outputs=gallery)
57
 
58
  generate_btn.click(fn=generate, inputs=[prompt_input, num_images_input, guidance_scale], outputs=gallery)
59
-
 
60
 
61
  demo.queue().launch()
 
34
 
35
  with gr.Blocks() as demo:
36
  gr.Markdown("# SPIN-Diffusion 1.0 Demo")
37
+ gr.Markdown("A self-play fine-tuned diffusion model from runwayml/stable-diffusion-v1-5, using winner images from the yuvalkirstain/pickapic_v2 dataset.")
38
  with gr.Row():
39
  prompt_input = gr.Textbox(label="Enter your prompt", placeholder="Type something...", lines=2)
40
  generate_btn = gr.Button("Generate images")
 
56
  gr.Examples(examples=examples, inputs=prompt_input, fn=generate, outputs=gallery)
57
 
58
  generate_btn.click(fn=generate, inputs=[prompt_input, num_images_input, guidance_scale], outputs=gallery)
59
+
60
+ gr.Markdown("If a generated image appears entirely black, it has been filtered out by the NSFW safety checker. Please try generating additional images.)
61
 
62
  demo.queue().launch()