search

dmarx / video-killed-the-radio-star

Notebook and tools for end-to-end automation of music video production with generative AI
https://colab.research.google.com/github/dmarx/video-killed-the-radio-star/blob/main/Video_Killed_The_Radio_Star_Defusion.ipynb#scrollTo=oPbeyWtesAoh
MIT License
196 stars 35 forks source link

Adds Parameter For Regenerating NSFW Images #122

Closed SchmitzAndrew closed 1 year ago

SchmitzAndrew commented 1 year ago

This PR is in response to Issue #104.

I added a parameter to allow the user to regenerate the image if it gets marked as NSFW. My solution detects whether or not the image is NSFW by determining if the returned image is solid black.

It regenerates the image a set amount of times as determined by the user.