Spaces:
Build error
Build error
Commit
·
033e75d
1
Parent(s):
cc7ee18
Update app.py
Browse files
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 |
-
|
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 |
|