Update app.py

app.py

@@ -49,10 +49,12 @@ def infer(
 
     generator = torch.Generator().manual_seed(seed)
     
-    pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch_dtype)
+    pipe = None
     if (model_repo_id=="stable-diffusion-v1-5/stable-diffusion-v1-5"): 
-        pipe.unet = PeftModel.from_pretrained(pipe.unet,"um235/cartoon_cat_stickers")
-        pipe.scale_lora(lscale)
+        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})
+    else:
+        pipe=DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch_dtype)
     pipe = pipe.to(device)
 
     image = pipe(