Hugging Face's logo

Spaces:
playgroundai
/
playground-v2.5
Running on Zero

App Files Community
9
ehsanakh commited on
Commit
1fa0c6b
•
1 Parent(s): b5e043c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,7 +11,7 @@ import spaces
11
  import torch
12
  from diffusers import DiffusionPipeline
13
 
14
- DESCRIPTION = """# Playground v2"""
15
  if not torch.cuda.is_available():
16
  DESCRIPTION += "\n<p>Running on CPU 🥶 This demo may not work on CPU.</p>"
17
 
@@ -27,7 +27,7 @@ NUM_IMAGES_PER_PROMPT = 1
27
 
28
  if torch.cuda.is_available():
29
  pipe = DiffusionPipeline.from_pretrained(
30
- "playgroundai/playground-v2-1024px-aesthetic",
31
  torch_dtype=torch.float16,
32
  use_safetensors=True,
33
  add_watermarker=False,
 
11
  import torch
12
  from diffusers import DiffusionPipeline
13
 
14
+ DESCRIPTION = """# Playground v2.5"""
15
  if not torch.cuda.is_available():
16
  DESCRIPTION += "\n<p>Running on CPU 🥶 This demo may not work on CPU.</p>"
17
 
 
27
 
28
  if torch.cuda.is_available():
29
  pipe = DiffusionPipeline.from_pretrained(
30
+ "playgroundai/playground-v2.5-1024px-aesthetic",
31
  torch_dtype=torch.float16,
32
  use_safetensors=True,
33
  add_watermarker=False,