Update app.py

app.py

@@ -8,7 +8,6 @@ from diffusers import DiffusionPipeline
 import torch
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-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
 else:
@@ -45,10 +44,11 @@ def infer(
     generator = torch.Generator().manual_seed(seed)
     
     pipe = None
-    if (model_repo_id=="stable-diffusion-v1-5/stable-diffusion-v1-5 with lora"): 
+    if (model_id=="stable-diffusion-v1-5/stable-diffusion-v1-5 with lora"): 
         pipe=DiffusionPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5", torch_dtype=torch_dtype,cross_attention_kwargs={"scale": 0.5})
         pipe.unet = PeftModel.from_pretrained(pipe.unet,"um235/cartoon_cat_stickers")
     else:
+        print("stable-diffusion-v1-5/stable-diffusion-v1-5 with lora")
         pipe=DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch_dtype)
     pipe = pipe.to(device)