MuhammadHanif commited on
Commit
ee4fae6
1 Parent(s): 521234d

whoops forgot to extract image from list

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -42,7 +42,7 @@ def infer(prompts, negative_prompts, width=1088, height=1088, inference_steps=30
42
  ).images
43
 
44
  output_images = pipe.numpy_to_pil(np.asarray(output.reshape((num_samples,) + output.shape[-3:])))
45
- return output_images
46
 
47
  prompt_input = gr.inputs.Textbox(
48
  label="Prompt",
@@ -54,11 +54,11 @@ neg_prompt_input = gr.inputs.Textbox(
54
  )
55
 
56
  width_slider = gr.inputs.Slider(
57
- minimum=512, maximum=2048, default=30, step=64, label="width"
58
  )
59
 
60
  height_slider = gr.inputs.Slider(
61
- minimum=512, maximum=2048, default=30, step=64, label="height"
62
  )
63
 
64
  inf_steps_input = gr.inputs.Slider(
 
42
  ).images
43
 
44
  output_images = pipe.numpy_to_pil(np.asarray(output.reshape((num_samples,) + output.shape[-3:])))
45
+ return output_images[0]
46
 
47
  prompt_input = gr.inputs.Textbox(
48
  label="Prompt",
 
54
  )
55
 
56
  width_slider = gr.inputs.Slider(
57
+ minimum=512, maximum=2048, default=1088, step=64, label="width"
58
  )
59
 
60
  height_slider = gr.inputs.Slider(
61
+ minimum=512, maximum=2048, default=1088, step=64, label="height"
62
  )
63
 
64
  inf_steps_input = gr.inputs.Slider(