Update app.py

app.py

@@ -15,8 +15,16 @@ if torch.cuda.is_available():
 else:
     torch_dtype = torch.float32
 
-pipe = StableDiffusionXLPipeline.from_pretrained(model_repo_id, vae=vae, torch_dtype=torch_dtype, use_safetensors=True)
-pipe = pipe.to(device)
+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",
+    )
+pipe.to(device)
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 4096