StevenTGay commited on
Commit
27271eb
1 Parent(s): 89932ef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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,