Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
use gr.JSON to send array
Browse files
app.py
CHANGED
@@ -37,15 +37,12 @@ def infer(prompt, samples, steps, scale, seed, width):
|
|
37 |
)
|
38 |
images = []
|
39 |
safe_image = Image.open(r"unsafe.png")
|
40 |
-
contains_nfsw = []
|
41 |
for i, image in enumerate(images_list["sample"]):
|
42 |
if(images_list["nsfw_content_detected"][i]):
|
43 |
images.append(safe_image)
|
44 |
-
contains_nfsw.append([True])
|
45 |
else:
|
46 |
images.append(image)
|
47 |
-
|
48 |
-
return [images, contains_nfsw]
|
49 |
|
50 |
css = """
|
51 |
.gradio-container {
|
@@ -276,7 +273,7 @@ with block:
|
|
276 |
step=1,
|
277 |
randomize=True,
|
278 |
)
|
279 |
-
contains_nfsw = gr.
|
280 |
width = gr.Number(value=128, visible=False)
|
281 |
ex = gr.Examples(examples=examples, fn=infer, inputs=[text, samples, steps, scale, seed, width], outputs=[gallery, contains_nfsw], cache_examples=False)
|
282 |
ex.dataset.headers = [""]
|
|
|
37 |
)
|
38 |
images = []
|
39 |
safe_image = Image.open(r"unsafe.png")
|
|
|
40 |
for i, image in enumerate(images_list["sample"]):
|
41 |
if(images_list["nsfw_content_detected"][i]):
|
42 |
images.append(safe_image)
|
|
|
43 |
else:
|
44 |
images.append(image)
|
45 |
+
return [images, images_list["nsfw_content_detected"]]
|
|
|
46 |
|
47 |
css = """
|
48 |
.gradio-container {
|
|
|
273 |
step=1,
|
274 |
randomize=True,
|
275 |
)
|
276 |
+
contains_nfsw = gr.JSON()
|
277 |
width = gr.Number(value=128, visible=False)
|
278 |
ex = gr.Examples(examples=examples, fn=infer, inputs=[text, samples, steps, scale, seed, width], outputs=[gallery, contains_nfsw], cache_examples=False)
|
279 |
ex.dataset.headers = [""]
|