Spaces:
Sleeping
Sleeping
saranbalan
commited on
Commit
•
64121ed
1
Parent(s):
1f5ca02
Update app.py
Browse files
app.py
CHANGED
@@ -18,7 +18,7 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
|
18 |
whisper_model = whisper.load_model("base")
|
19 |
|
20 |
# Model IDs for Stable Diffusion pipelines
|
21 |
-
HF_API_KEY = https://huggingface.co/ByteDance/SDXL-Lightning/resolve/main/sdxl_lightning_1step_x0.safetensors
|
22 |
|
23 |
# Initialize Stable Diffusion pipeline based on device
|
24 |
if torch.cuda.is_available():
|
|
|
18 |
whisper_model = whisper.load_model("base")
|
19 |
|
20 |
# Model IDs for Stable Diffusion pipelines
|
21 |
+
HF_API_KEY = "https://huggingface.co/ByteDance/SDXL-Lightning/resolve/main/sdxl_lightning_1step_x0.safetensors"
|
22 |
|
23 |
# Initialize Stable Diffusion pipeline based on device
|
24 |
if torch.cuda.is_available():
|