Jackflack09 commited on
Commit
033e75d
·
1 Parent(s): cc7ee18

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -239,7 +239,7 @@ def img_to_img(model_path, prompt, n_images, neg_prompt, img, strength, guidance
239
 
240
  return replace_nsfw_images(result)
241
 
242
- def replace_nsfw_images(results):
243
 
244
  if is_colab:
245
  return results.images
@@ -251,7 +251,6 @@ def img_to_img(model_path, prompt, n_images, neg_prompt, img, strength, guidance
251
  nsfw_label = f"NSFW: {results.nsfw_content_prob[i]:.2f}"
252
  img_array = np.concatenate([np.array(orig_img), np.array(results.images[i])], axis=1)
253
  display(Image.fromarray(img_array), caption=nsfw_label)
254
-
255
  return results.images
256
 
257
 
 
239
 
240
  return replace_nsfw_images(result)
241
 
242
+ def replace_nsfw_images(results):
243
 
244
  if is_colab:
245
  return results.images
 
251
  nsfw_label = f"NSFW: {results.nsfw_content_prob[i]:.2f}"
252
  img_array = np.concatenate([np.array(orig_img), np.array(results.images[i])], axis=1)
253
  display(Image.fromarray(img_array), caption=nsfw_label)
 
254
  return results.images
255
 
256