Update app.py

Removed SD 1.5

app.py

@@ -34,11 +34,6 @@ def load_model(model_name):
             torch_dtype=torch.float16, 
             variant="fp16"
         ).to("cuda")
-    elif model_name == "runwayml/stable-diffusion-v1-5":
-        pipeline = StableDiffusionPipeline.from_pretrained(
-            model_name, 
-            torch_dtype=torch.float16
-        ).to("cuda")
     elif model_name == "ByteDance/SDXL-Lightning":
         base = "stabilityai/stable-diffusion-xl-base-1.0"
         ckpt = "sdxl_lightning_4step_unet.safetensors"
@@ -82,8 +77,6 @@ pipeline_text2image = load_model(default_model)
 def getimgen(prompt, model_name):
     if model_name == "stabilityai/sdxl-turbo":
         return pipeline_text2image(prompt=prompt, guidance_scale=0.0, num_inference_steps=2).images[0]
-    elif model_name == "runwayml/stable-diffusion-v1-5":
-        return pipeline_text2image(prompt).images[0]
     elif model_name == "ByteDance/SDXL-Lightning":
         return pipeline_text2image(prompt, num_inference_steps=4, guidance_scale=0).images[0]
     elif model_name == "segmind/SSD-1B":
@@ -273,7 +266,6 @@ This demo provides an insightful look into how current text-to-image models hand
             "stabilityai/sdxl-turbo", 
             "ByteDance/SDXL-Lightning",
             "stabilityai/stable-diffusion-2",
-            "runwayml/stable-diffusion-v1-5", 
             "segmind/SSD-1B"
         ], 
         value=default_model