Spaces:
Runtime error
Runtime error
Fabrice-TIERCELIN
commited on
Commit
•
94299d1
1
Parent(s):
63a3f35
Handle Negative prompt
Browse filesThis PR allows the user to tell what we do not want on the image.
_Successfully tested here:_ https://huggingface.co/spaces/Fabrice-TIERCELIN/SDXL-Turbo-Realtime
Click on _Merge_ to approve
app.py
CHANGED
@@ -5,14 +5,17 @@ from diffusers import DiffusionPipeline
|
|
5 |
pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo")
|
6 |
|
7 |
@spaces.GPU(duration=250)
|
8 |
-
def generate_image(prompt):
|
9 |
# Run the diffusion model to generate an image
|
10 |
-
output = pipe(prompt, num_inference_steps=50, guidance_scale=7.5)
|
11 |
return output.images[0]
|
12 |
|
|
|
|
|
|
|
13 |
gr_interface = gr.Interface(
|
14 |
fn=generate_image,
|
15 |
-
inputs=
|
16 |
outputs="image",
|
17 |
title="Real-time Image Generation with Diffusion",
|
18 |
description="Enter a prompt to generate an image",
|
|
|
5 |
pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo")
|
6 |
|
7 |
@spaces.GPU(duration=250)
|
8 |
+
def generate_image(prompt, negative_prompt):
|
9 |
# Run the diffusion model to generate an image
|
10 |
+
output = pipe(prompt, negative_prompt, num_inference_steps=50, guidance_scale=7.5)
|
11 |
return output.images[0]
|
12 |
|
13 |
+
prompt = gr.Textbox(label = "Prompt", info = "Describe the subject, the background and the style of image; 77 token limit", placeholder = "Describe what you want to see", lines = 2)
|
14 |
+
negative_prompt = gr.Textbox(label = "Negative prompt", placeholder = "Describe what you do NOT want to see", value = "Ugly, malformed, noise, blur, watermark")
|
15 |
+
|
16 |
gr_interface = gr.Interface(
|
17 |
fn=generate_image,
|
18 |
+
inputs=[prompt, negative_prompt],
|
19 |
outputs="image",
|
20 |
title="Real-time Image Generation with Diffusion",
|
21 |
description="Enter a prompt to generate an image",
|