Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -4,7 +4,7 @@ from diffusers import StableDiffusionPipeline
|
|
| 4 |
# Load the Hugging Face model
|
| 5 |
model_id = "kopyl/ui-icons-256"
|
| 6 |
pipeline = StableDiffusionPipeline.from_pretrained(model_id)
|
| 7 |
-
pipeline.to("cuda") # Use GPU for faster inference, if available
|
| 8 |
|
| 9 |
def generate_icon(prompt):
|
| 10 |
"""Generate an icon from a text prompt."""
|
|
|
|
| 4 |
# Load the Hugging Face model
|
| 5 |
model_id = "kopyl/ui-icons-256"
|
| 6 |
pipeline = StableDiffusionPipeline.from_pretrained(model_id)
|
| 7 |
+
#pipeline.to("cuda") # Use GPU for faster inference, if available
|
| 8 |
|
| 9 |
def generate_icon(prompt):
|
| 10 |
"""Generate an icon from a text prompt."""
|