Update app.py

app.py

@@ -46,7 +46,8 @@ def infer(
     pipe = None
     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)
-        pipe.unet = PeftModel.from_pretrained(pipe.unet,"um235/cartoon_cat_stickers")
+        pipe.unet = PeftModel.from_pretrained(pipe.unet,"um235/cartoon_cat_stickers/unet")
+        pipe.text_encoder= PeftModel.from_pretrained(pipe.text_encoder,"um235/cartoon_cat_stickers/text_encoder")
     else:
         print("stable-diffusion-v1-5/stable-diffusion-v1-5 with lora")
         pipe=DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch_dtype)
@@ -69,7 +70,7 @@ def infer(
 examples = [
     "Sticker VanillaCat. Cartoon-style cat with soft yellow fur and a white flower on its head, sitting up with a relaxed expression, eyes half-closed, content and calm, casual pose, peaceful mood, white background.",
     "Sticker VanillaCat. Cartoon-style cat with soft yellow fur and a white flower on its head, standing with a mischievous grin, one paw raised playfully, bright eyes full of energy, cheeky and fun, white background",
-    "VanillaCat. Cartoon-style cat with soft yellow fur and a white flower on its head, jumping mid-air with a surprised expression, wide eyes, and mouth open in excitement, paws stretched out, energetic and playful, white background.",
+    "Sticker VanillaCat. Cartoon-style cat with soft yellow fur and a white flower on its head, jumping mid-air with a surprised expression, wide eyes, and mouth open in excitement, paws stretched out, energetic and playful, forest background.",
 ]
 
 css = """