radames commited on
Commit
f25aa76
1 Parent(s): b60caa9

use gr.JSON to send array

Browse files
Files changed (1) hide show
  1. app.py +2 -5
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
- contains_nfsw.append([False])
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.Dataframe(type="numpy", datatype="bool", visible=False)
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 = [""]