justinpinkney commited on
Commit
0b6171d
1 Parent(s): b08b4dd

update seed and image loading

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -21,9 +21,9 @@ def main(
21
  )
22
 
23
  images = []
24
- safe_image = Image.open(r"unsafe.png")
25
  for i, image in enumerate(images_list["sample"]):
26
  if(images_list["nsfw_content_detected"][i]):
 
27
  images.append(safe_image)
28
  else:
29
  images.append(image)
@@ -66,7 +66,7 @@ inputs = [
66
  gr.Slider(0, 25, value=3, step=1, label="Guidance scale"),
67
  gr.Slider(1, 4, value=1, step=1, label="Number images"),
68
  gr.Slider(5, 50, value=25, step=5, label="Steps"),
69
- gr.Number(0, 1024, value=0, step=1, labal="Seed")
70
  ]
71
  output = gr.Gallery(label="Generated variations")
72
  output.style(grid=2)
 
21
  )
22
 
23
  images = []
 
24
  for i, image in enumerate(images_list["sample"]):
25
  if(images_list["nsfw_content_detected"][i]):
26
+ safe_image = Image.open(r"unsafe.png")
27
  images.append(safe_image)
28
  else:
29
  images.append(image)
 
66
  gr.Slider(0, 25, value=3, step=1, label="Guidance scale"),
67
  gr.Slider(1, 4, value=1, step=1, label="Number images"),
68
  gr.Slider(5, 50, value=25, step=5, label="Steps"),
69
+ gr.Number(0, labal="Seed", precision=0)
70
  ]
71
  output = gr.Gallery(label="Generated variations")
72
  output.style(grid=2)