Update app.py

app.py

@@ -18,14 +18,13 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 whisper_model = whisper.load_model("base")
 
 # Model IDs for Stable Diffusion pipelines
-model_id1 = "dreamlike-art/dreamlike-diffusion-1.0"
-model_id2 = "stabilityai/stable-diffusion-xl-base-1.0"
+ HF_API_KEY = https://huggingface.co/ByteDance/SDXL-Lightning/resolve/main/sdxl_lightning_1step_x0.safetensors
 
 # Initialize Stable Diffusion pipeline based on device
 if torch.cuda.is_available():
-    pipe = StableDiffusionPipeline.from_pretrained(model_id1, torch_dtype=torch.float16)
+    pipe = StableDiffusionPipeline.from_pretrained(HF_API_KEY, torch_dtype=torch.float16)
 else:
-    pipe = StableDiffusionPipeline.from_pretrained(model_id1)  # Omit torch_dtype for CPU
+    pipe = StableDiffusionPipeline.from_pretrained(HF_API_KEY)  # Omit torch_dtype for CPU
 
 # Move model to the selected device (either GPU or CPU)
 pipe = pipe.to(device)