Update app.py

app.py

@@ -51,8 +51,8 @@ def infer(
     
     pipe = None
     if (model_repo_id=="stable-diffusion-v1-5/stable-diffusion-v1-5 with lora"): 
-        pipe=DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch_dtype)
-        pipe.unet = PeftModel.from_pretrained(pipe.unet,"um235/cartoon_cat_stickers",cross_attention_kwargs={"scale": 0.5})
+        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:
         pipe=DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch_dtype)
     pipe = pipe.to(device)