RealTimeImageGen

Runtime error

awacke1 commited on Nov 23, 2023

Commit

feb562f

•

1 Parent(s): 660eabc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -78,8 +78,9 @@ def predict(prompt, guidance, steps, seed=1231231):
         else False
     )
     if nsfw_content_detected:
-        gr.Warning("NSFW content detected.")
-        return Image.new("RGB", (512, 512))
     return results.images[0]

         else False
     )
     if nsfw_content_detected:
+        nsfw=gr.Button("🕹️NSFW🎨", scale=1)
+    #    gr.Warning("NSFW content detected.")
+    #    return Image.new("RGB", (512, 512))
     return results.images[0]