jschen commited on
Commit
0fe18e7
1 Parent(s): 2cf3dde

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -5
app.py CHANGED
@@ -13,9 +13,10 @@ import torch
13
 
14
  from diffusers import AutoencoderKL, PixArtAlphaPipeline
15
 
16
- DESCRIPTION = """# PixArt-Alpha 1024
17
- #### [PixArt-Alpha 1024](https://github.com/PixArt-alpha/PixArt-alpha) is a transformer-based text-to-image diffusion system trained on text embeddings from T5. This demo uses the [PixArt-alpha/PixArt-XL-2-1024-MS](https://huggingface.co/PixArt-alpha/PixArt-XL-2-1024-MS) checkpoint.
18
- """
 
19
  if not torch.cuda.is_available():
20
  DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>"
21
 
@@ -82,7 +83,7 @@ style_list = [
82
 
83
  styles = {k["name"]: (k["prompt"], k["negative_prompt"]) for k in style_list}
84
  STYLE_NAMES = list(styles.keys())
85
- DEFAULT_STYLE_NAME = "Cinematic"
86
 
87
 
88
  def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str, str]:
@@ -164,12 +165,14 @@ def generate(
164
 
165
 
166
  examples = [
 
 
 
167
  "3d digital art of an adorable ghost, glowing within, holding a heart shaped pumpkin, Halloween, super cute, spooky haunted house background",
168
  "beautiful lady, freckles, big smile, blue eyes, short ginger hair, dark makeup, wearing a floral blue vest top, soft light, dark grey background",
169
  "professional portrait photo of an anthropomorphic cat wearing fancy gentleman hat and jacket walking in autumn forest.",
170
  "an astronaut sitting in a diner, eating fries, cinematic, analog film",
171
  "Albert Einstein in a surrealist Cyberpunk 2077 world, hyperrealistic",
172
- "cinematic film still of Futuristic hero with golden dark armour with machine gun, muscular body",
173
  ]
174
 
175
  with gr.Blocks(css="style.css") as demo:
 
13
 
14
  from diffusers import AutoencoderKL, PixArtAlphaPipeline
15
 
16
+ DESCRIPTION = """![Logo](https://raw.githubusercontent.com/PixArt-alpha/PixArt-alpha.github.io/master/static/images/logo.png)
17
+ # PixArt-Alpha 1024
18
+ #### [PixArt-Alpha 1024](https://github.com/PixArt-alpha/PixArt-alpha) is a transformer-based text-to-image diffusion system trained on text embeddings from T5. This demo uses the [PixArt-alpha/PixArt-XL-2-1024-MS](https://huggingface.co/PixArt-alpha/PixArt-XL-2-1024-MS) checkpoint.
19
+ """
20
  if not torch.cuda.is_available():
21
  DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>"
22
 
 
83
 
84
  styles = {k["name"]: (k["prompt"], k["negative_prompt"]) for k in style_list}
85
  STYLE_NAMES = list(styles.keys())
86
+ DEFAULT_STYLE_NAME = "(No style)"
87
 
88
 
89
  def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str, str]:
 
165
 
166
 
167
  examples = [
168
+ "A small cactus with a happy face in the Sahara desert.",
169
+ "Pirate ship trapped in a cosmic maelstrom nebula, rendered in cosmic beach whirlpool engine, volumetric lighting, spectacular, ambient lights, light pollution, cinematic atmosphere, art nouveau style, illustration art artwork by SenseiJaye, intricate detail.",
170
+ "stars, water, brilliantly, gorgeous large scale scene, a little girl, in the style of dreamy realism, light gold and amber, blue and pink, brilliantly illuminated in the background.",
171
  "3d digital art of an adorable ghost, glowing within, holding a heart shaped pumpkin, Halloween, super cute, spooky haunted house background",
172
  "beautiful lady, freckles, big smile, blue eyes, short ginger hair, dark makeup, wearing a floral blue vest top, soft light, dark grey background",
173
  "professional portrait photo of an anthropomorphic cat wearing fancy gentleman hat and jacket walking in autumn forest.",
174
  "an astronaut sitting in a diner, eating fries, cinematic, analog film",
175
  "Albert Einstein in a surrealist Cyberpunk 2077 world, hyperrealistic",
 
176
  ]
177
 
178
  with gr.Blocks(css="style.css") as demo: