Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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()
|