Commit
·
0b6171d
1
Parent(s):
b08b4dd
update seed and image loading
Browse files
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,
|
| 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)
|