Update app.py

app.py

@@ -16,7 +16,7 @@ else:
 
 pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
 #pipe.unet.load_adapter("um235/cartoon_cat_stickers")
-pipe.unet = PeftModel.from_pretrained(pipe.unet,"um235/cartoon_cat_stickers")
+#pipe.unet = PeftModel.from_pretrained(pipe.unet,"um235/cartoon_cat_stickers")
 pipe = pipe.to(device)
 
 MAX_SEED = np.iinfo(np.int32).max
@@ -42,7 +42,7 @@ def infer(
     generator = torch.Generator().manual_seed(seed)
     
     pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch_dtype)
-    #if (model_repo_id=="stable-diffusion-v1-5/stable-diffusion-v1-5"): pipe.unet = PeftModel.from_pretrained(pipe.unet,"um235/cartoon_cat_stickers")
+    if (model_repo_id=="stable-diffusion-v1-5/stable-diffusion-v1-5"): pipe.unet = PeftModel.from_pretrained(pipe.unet,"um235/cartoon_cat_stickers")
     pipe = pipe.to(device)
 
     image = pipe(