Chris4K commited on
Commit
f1965d0
·
verified ·
1 Parent(s): 0db9f5d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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."""