Update app.py
Browse files
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)
|