Update app.py

app.py

@@ -246,8 +246,8 @@ MAX_SEED = np.iinfo(np.int32).max
 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-v150-sdxl",torch_dtype=torch.float16)
+    # vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)    
+    pipe = StableDiffusionXLPipeline.from_pretrained("John6666/rrod-ikea-rrod-ikea-v5-sdxl",torch_dtype=torch.float16)
     # pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
     pipe.to("cuda")
 
@@ -313,9 +313,9 @@ def infer(
         "height": height,
         "guidance_scale": guidance_scale,
         "num_inference_steps": num_inference_steps,
-        "model": "wai-nsfw-illustrious-sdxl-v150-sdxl",
+        "model": "rrod-ikea-rrod-ikea-v5-sdxl",
         "use_resolution_binning": use_resolution_binning,
-        "PreUrl": "https://huggingface.co/spaces/Menyu/wainsfw"
+        "PreUrl": "https://huggingface.co/spaces/asffhgjasdasfdsa/spaceTest"
     }
     # Add metadata to the image
     image_with_metadata = add_metadata_to_image(image, metadata)
@@ -336,9 +336,7 @@ css = '''
 h1{text-align:center}
 '''
 
-with gr.Blocks(css=css) as demo:
-    gr.Markdown("""# 梦羽的模型生成器
-        ### 快速生成 Wai nsfw illustrious v150 模型的图片""")
+with gr.Blocks(css=css) as demo:    
     with gr.Group():
         with gr.Row():
             prompt = gr.Text(