Update app.py

app.py

@@ -247,7 +247,7 @@ MAX_IMAGE_SIZE = 2048
 
 if torch.cuda.is_available():
     # vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
-    pipe = StableDiffusionXLPipeline.from_pretrained("John6666/wai-nsfw-illustrious-sdxl-v140-sdxl",torch_dtype=torch.float16)
+    pipe = StableDiffusionXLPipeline.from_pretrained("John6666/wai-nsfw-illustrious-sdxl-v150-sdxl",torch_dtype=torch.float16)
     # pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
     pipe.to("cuda")
 
@@ -313,7 +313,7 @@ def infer(
         "height": height,
         "guidance_scale": guidance_scale,
         "num_inference_steps": num_inference_steps,
-        "model": "wai-nsfw-illustrious-sdxl-v140-sdxl",
+        "model": "wai-nsfw-illustrious-sdxl-v150-sdxl",
         "use_resolution_binning": use_resolution_binning,
         "PreUrl": "https://huggingface.co/spaces/Menyu/wainsfw"
     }
@@ -338,7 +338,7 @@ h1{text-align:center}
 
 with gr.Blocks(css=css) as demo:
     gr.Markdown("""# 梦羽的模型生成器
-        ### 快速生成 Wai nsfw illustrious v140 模型的图片""")
+        ### 快速生成 Wai nsfw illustrious v150 模型的图片""")
     with gr.Group():
         with gr.Row():
             prompt = gr.Text(