Update app.py (#5)

- Update app.py (03b327763577e2d8e127af844ea9e040a784adbb)


Co-authored-by: yoinked <yoinked@users.noreply.huggingface.co>

app.py

@@ -2,7 +2,7 @@ import spaces
 import os
 import gc
 import gradio as gr
-import gradio_client as grcl
+# import gradio_client as grcl
 import numpy as np
 import torch
 import json
@@ -13,7 +13,7 @@ from PIL import Image, PngImagePlugin
 from datetime import datetime
 from diffusers.models import AutoencoderKL
 from diffusers import StableDiffusionXLPipeline, StableDiffusionXLImg2ImgPipeline
-GRAD_CLIENT = grcl.Client("https://yoinked-da-nsfw-checker.hf.space/")
+#GRAD_CLIENT = grcl.Client("https://yoinked-da-nsfw-checker.hf.space/")
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
@@ -83,7 +83,11 @@ def generate(
     upscale_by: float = 1.5,
     add_quality_tags: bool = True,
     progress=gr.Progress(track_tqdm=True),
+    nsfw_neg=True
 ):
+    if nsfw_neg:
+        prompt += "general, "
+        negative_prompt += ", explicit, questionable, nude, naked, pussy, penis, uncensored" # mikudayo
     generator = utils.seed_everything(seed)
 
     width, height = utils.aspect_ratio_handler(
@@ -195,16 +199,16 @@ def generate(
 
 def genwrap(*args, **kwargs):
     ipth, mtd = generate(*args, **kwargs)
-    r = GRAD_CLIENT.predict(ipth, "chen-evangelion", 0.4, False, False, api_name="/classify")
-    ratings = val[0]
-    rating = rating['confidences']
-    highestval, classtype = -1, "aa"
-    for o in rating:
-        if o['confidence'] > highestval:
-            highestval = o['confidence']
-            classtype = o['label']
-    if classtype not in ["general", "sensitive"]: #add "questionable" and "explicit" to enable nsfw, or just delete this func
-        return "https://upload.wikimedia.org/wikipedia/commons/b/bf/Bucephala-albeola-010.jpg", mtd
+    #r = GRAD_CLIENT.predict(ipth, "chen-evangelion", 0.4, False, False, api_name="/classify")
+    #ratings = val[0]
+    #rating = rating['confidences']
+    #highestval, classtype = -1, "aa"
+    #for o in rating:
+    #    if o['confidence'] > highestval:
+    #        highestval = o['confidence']
+    #        classtype = o['label']
+    #if classtype not in ["general", "sensitive"]: # i hate code
+    #    return "https://upload.wikimedia.org/wikipedia/commons/b/bf/Bucephala-albeola-010.jpg", mtd
     return ipth, mtd
 if torch.cuda.is_available():
     pipe = load_pipeline(MODEL)
@@ -336,6 +340,8 @@ with gr.Blocks(css="style.css", theme="NoCrypt/miku@1.2.1") as demo:
                             step=1,
                             value=28,
                         )
+                with gr.Accordion(label="lewdie.... lewd lewdie...", open=False):
+                    nsfwtoggle = gr.Checkbox(label="Anti-NSFW [dont disable this if korean]", value=True)
         with gr.Column(scale=3):
             with gr.Blocks():
                 run_button = gr.Button("Generate", variant="primary")
@@ -399,6 +405,7 @@ with gr.Blocks(css="style.css", theme="NoCrypt/miku@1.2.1") as demo:
             upscaler_strength,
             upscale_by,
             add_quality_tags,
+            nsfwtoggle
         ],
         outputs=[result, gr_metadata],
         api_name="run",