Raphael commited on
Commit
22587f6
1 Parent(s): de3e780

Fix nsfw black background

Browse files

Signed-off-by: Raphael <oOraph@users.noreply.github.com>

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -94,7 +94,7 @@ def generate_background(prompt, num_inference_steps, height, width):
94
 
95
  if nsfw:
96
  LOG.info('NSFW detected, skipping')
97
- background = np.zeros((height, width), dtype='uint8')
98
  else:
99
  background = np.array(background)
100
  # Convert RGB to BGR
 
94
 
95
  if nsfw:
96
  LOG.info('NSFW detected, skipping')
97
+ background = np.zeros((height, width, 3), dtype='uint8')
98
  else:
99
  background = np.array(background)
100
  # Convert RGB to BGR