Update app.py

app.py

@@ -7,7 +7,7 @@ from diffusers import DiffusionPipeline
 import torch
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model_repo_id = "CompVis/stable-diffusion-v1-4"  # Replace to the model you would like to use
+model_repo_id = "stable-diffusion-v1-5/stable-diffusion-v1-5"  # Replace to the model you would like to use
 
 if torch.cuda.is_available():
     torch_dtype = torch.float16
@@ -15,6 +15,7 @@ else:
     torch_dtype = torch.float32
 
 pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
+pipe.unet.load_attn_procs("juliensimon/stable-diffusion-v1-5-pokemon-lora")
 pipe = pipe.to(device)
 
 MAX_SEED = np.iinfo(np.int32).max
@@ -40,6 +41,7 @@ def infer(
     generator = torch.Generator().manual_seed(seed)
     
     pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch_dtype)
+    if (model_repo_id=="stable-diffusion-v1-5/stable-diffusion-v1-5"): pipe.unet.load_attn_procs("juliensimon/stable-diffusion-v1-5-pokemon-lora")
     pipe = pipe.to(device)
 
     image = pipe(