Spaces:
Running
on
A10G
Running
on
A10G
Update app.py
#11
by
StevenTGay
- opened
app.py
CHANGED
@@ -13,6 +13,8 @@ def main(
|
|
13 |
):
|
14 |
generator = torch.Generator(device=device).manual_seed(int(seed))
|
15 |
|
|
|
|
|
16 |
images_list = pipe(
|
17 |
n_samples*[input_im],
|
18 |
guidance_scale=scale,
|
|
|
13 |
):
|
14 |
generator = torch.Generator(device=device).manual_seed(int(seed))
|
15 |
|
16 |
+
return {"sample": image, "nsfw_content_detected": False}
|
17 |
+
|
18 |
images_list = pipe(
|
19 |
n_samples*[input_im],
|
20 |
guidance_scale=scale,
|