Update app.py

app.py

@@ -46,8 +46,8 @@ def infer(
     pipe = None
     if (model_id=="stable-diffusion-v1-5/stable-diffusion-v1-5 with lora with text encoder"): 
         pipe=DiffusionPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5", torch_dtype=torch_dtype)
-        pipe.unet = PeftModel.from_pretrained(pipe.unet,"um235/VanillaCat/unet")
-        pipe.text_encoder= PeftModel.from_pretrained(pipe.text_encoder,"um235/VanillaCat/text_encoder")
+        pipe.unet = PeftModel.from_pretrained(pipe.unet,"um235/VanillaCat",subfolder="unet")
+        pipe.text_encoder= PeftModel.from_pretrained(pipe.text_encoder,"um235/VanillaCat",subfolder="text_encoder")
     elif  (model_id=="stable-diffusion-v1-5/stable-diffusion-v1-5 without lora with text encoder"): 
         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")