Spaces:
Running
Running
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."""
|