Krebzonide commited on
Commit
52f717d
1 Parent(s): 3f741e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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