fffiloni commited on
Commit
c626e5c
β€’
1 Parent(s): 86ddc39

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,18 +14,18 @@ def ask(input_im, scale, n_samples, steps, seed, images):
14
  num_inference_steps=steps,
15
  generator=generator,
16
  )
 
17
  for i, image in enumerate(images_list["sample"]):
18
  if(images_list["nsfw_content_detected"][i]):
19
  safe_image = Image.open(r"unsafe.png")
20
  images.append(safe_image)
21
  else:
22
  images.append(image)
 
23
  return images
24
 
25
  def main(input_im, scale, n_samples, steps, seed):
26
 
27
-
28
-
29
  images = []
30
  images = ask(input_im, scale, n_samples, steps, seed, images)
31
  images = ask(input_im, scale, n_samples, steps, seed, images)
 
14
  num_inference_steps=steps,
15
  generator=generator,
16
  )
17
+
18
  for i, image in enumerate(images_list["sample"]):
19
  if(images_list["nsfw_content_detected"][i]):
20
  safe_image = Image.open(r"unsafe.png")
21
  images.append(safe_image)
22
  else:
23
  images.append(image)
24
+
25
  return images
26
 
27
  def main(input_im, scale, n_samples, steps, seed):
28
 
 
 
29
  images = []
30
  images = ask(input_im, scale, n_samples, steps, seed, images)
31
  images = ask(input_im, scale, n_samples, steps, seed, images)