Krebzonide
commited on
Commit
•
52f717d
1
Parent(s):
3f741e5
Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ def generate( #Creates a function called "generate" that takes creates the image
|
|
14 |
progress=gr.Progress(track_tqdm=True)): #Shows you a progress bar as your image is generated
|
15 |
image = pipe( #The function that uses the pipeline to generate images
|
16 |
prompt, negative_prompt=negative_prompt, num_inference_steps=sampling_steps, guidance_scale=guidance_scale #The parameters that the pipeline will use
|
17 |
-
).images #Separates the image from the other data returned
|
18 |
return image #Sends the image back to whoever called this function
|
19 |
|
20 |
with gr.Blocks() as demo: #Creates a page on a website that you can look at and click on
|
|
|
14 |
progress=gr.Progress(track_tqdm=True)): #Shows you a progress bar as your image is generated
|
15 |
image = pipe( #The function that uses the pipeline to generate images
|
16 |
prompt, negative_prompt=negative_prompt, num_inference_steps=sampling_steps, guidance_scale=guidance_scale #The parameters that the pipeline will use
|
17 |
+
).images[0] #Separates the image from the other data returned
|
18 |
return image #Sends the image back to whoever called this function
|
19 |
|
20 |
with gr.Blocks() as demo: #Creates a page on a website that you can look at and click on
|