Update app.py

app.py

@@ -45,6 +45,10 @@ J9_pipe = StableDiffusionXLPipeline.from_pretrained(
         "RunDiffusion/Juggernaut-XL-v9",
         vae=vae,
         torch_dtype=torch.float16,
+        custom_pipeline="lpw_stable_diffusion_xl",
+        use_safetensors=True,
+        add_watermarker=False,
+        variant="fp16",
     )
 J9_pipe.to("cuda")
 
@@ -110,7 +114,7 @@ examples = ["A dignified beaver wearing glasses, a vest, and colorful neck tie."
 "A bald eagle made of chocolate powder, mango, and whipped cream"
 ]
 
-with gr.Blocks() as demo:
+with gr.Blocks(theme=gr.themes.Base()) as demo:
     gr.Markdown("## One step Juggernaut-XL comparison 🦶")
     gr.Markdown('Compare Juggernaut-XL variants and distillations able to generate images in a single diffusion step')
     prompt = gr.Textbox(label="Prompt")