Update app.py

app.py

@@ -44,12 +44,14 @@ def infer(
     generator = torch.Generator().manual_seed(seed)
     
     pipe = None
-    if (model_id=="stable-diffusion-v1-5/stable-diffusion-v1-5 with lora"): 
+    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/cartoon_cat_stickers/unet")
-        pipe.text_encoder= PeftModel.from_pretrained(pipe.text_encoder,"um235/cartoon_cat_stickers/text_encoder")
+        pipe.unet = PeftModel.from_pretrained(pipe.unet,"um235/VanillaCat/unet")
+        pipe.text_encoder= PeftModel.from_pretrained(pipe.text_encoder,"um235/VanillaCat/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")
     else:
-        print("stable-diffusion-v1-5/stable-diffusion-v1-5 with lora")
         pipe=DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch_dtype)
     pipe = pipe.to(device)
 
@@ -99,9 +101,10 @@ with gr.Blocks(css=css) as demo:
                 choices=[
                     "stable-diffusion-v1-5/stable-diffusion-v1-5",  
                     "CompVis/stable-diffusion-v1-4", 
-                    "stable-diffusion-v1-5/stable-diffusion-v1-5 with lora",  
+                    "stable-diffusion-v1-5/stable-diffusion-v1-5 with lora with text encoder",  
+                    "stable-diffusion-v1-5/stable-diffusion-v1-5 with lora without text encoder"
                 ],
-                value="stable-diffusion-v1-5/stable-diffusion-v1-5 with lora",  
+                value="stable-diffusion-v1-5/stable-diffusion-v1-5 with lora with text encoder",  
                 show_label=True,
                 type="value",  
         )