Update app.py

app.py

@@ -42,6 +42,7 @@ repo_customs = {
     "Default": repo_default,
     "Realistic": StableDiffusionXLPipeline.from_pretrained("stablediffusionapi/NightVision_XL", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=False, add_watermarker=False),
     "Anime": StableDiffusionXLPipeline.from_pretrained("cagliostrolab/animagine-xl-3.1", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
+    "Pixel": StableDiffusionXLPipeline.from_pretrained("PublicPrompts/All-In-One-Pixel-Model", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
 }
 
 # Functions
@@ -73,6 +74,9 @@ def generate(input=DEFAULT_INPUT, filter_input="", negative_input=DEFAULT_NEGATI
     elif model == "Anime":   
         steps = (not steps or steps < 0 and 16) or steps
         guidance = (not guidance or guidance < 0 and 7) or guidance
+    elif model == "Pixel":   
+        steps = (not steps or steps < 0 and 8) or steps
+        guidance = (not guidance or guidance < 0 and 3) or guidance
     else:
         steps = (not steps or steps < 0 and 16) or steps
         guidance = (not guidance or guidance < 0 and 3) or guidance
@@ -112,7 +116,7 @@ with gr.Blocks(css=css) as main:
         input = gr.Textbox(lines=1, value=DEFAULT_INPUT, label="Input")
         filter_input = gr.Textbox(lines=1, value="", label="Input Filter")
         negative_input = gr.Textbox(lines=1, value=DEFAULT_NEGATIVE_INPUT, label="Input Negative")
-        model = gr.Dropdown(label="Models", choices=["Default", "Realistic", "Anime"], value="Default")
+        model = gr.Dropdown(label="Models", choices=["Default", "Realistic", "Anime", "Pixel"], value="Default")
         height = gr.Slider(minimum=1, maximum=2160, step=1, value=DEFAULT_HEIGHT, label="Height")
         width = gr.Slider(minimum=1, maximum=2160, step=1, value=DEFAULT_WIDTH, label="Width")
         steps = gr.Slider(minimum=-1, maximum=100, step=1, value=-1, label="Steps")