ehristoforu commited on
Commit
19d1ef9
1 Parent(s): dc49fcd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -15,6 +15,7 @@ from diffusers.utils import load_image
15
  DESCRIPTION = """
16
  # [Fluently Playground](https://huggingface.co/fluently)
17
 
 
18
  """
19
  if not torch.cuda.is_available():
20
  DESCRIPTION += "\n<p>Running on CPU 🥶 This demo may not work on CPU.</p>"
@@ -68,7 +69,7 @@ if torch.cuda.is_available():
68
  pipe_inpaint.to(device)
69
 
70
  pipe_xl = StableDiffusionXLPipeline.from_pretrained(
71
- "fluently/Fluently-XL-v3",
72
  torch_dtype=torch.float16,
73
  use_safetensors=True,
74
  )
@@ -163,7 +164,7 @@ def generate(
163
  num_images_per_prompt=1,
164
  output_type="pil",
165
  ).images
166
- elif model == "Fluently XL v3":
167
  images = pipe_xl(
168
  prompt=prompt,
169
  negative_prompt=negative_prompt,
@@ -252,7 +253,7 @@ with gr.Blocks(title="Fluently Playground", css=css) as demo:
252
  with gr.Row():
253
  model = gr.Radio(
254
  label="Model",
255
- choices=["Fluently XL v3 Lightning","Fluently XL v3", "Fluently v4", "Fluently Anime", "Fluently Epic", "Fluently XL v3 inpaint", "Fluently v4 inpaint"],
256
  value="Fluently XL v3 Lightning",
257
  interactive=True,
258
  )
 
15
  DESCRIPTION = """
16
  # [Fluently Playground](https://huggingface.co/fluently)
17
 
18
+ 🤗 New FluentlyXL v4!
19
  """
20
  if not torch.cuda.is_available():
21
  DESCRIPTION += "\n<p>Running on CPU 🥶 This demo may not work on CPU.</p>"
 
69
  pipe_inpaint.to(device)
70
 
71
  pipe_xl = StableDiffusionXLPipeline.from_pretrained(
72
+ "fluently/Fluently-XL-v4",
73
  torch_dtype=torch.float16,
74
  use_safetensors=True,
75
  )
 
164
  num_images_per_prompt=1,
165
  output_type="pil",
166
  ).images
167
+ elif model == "Fluently XL v4":
168
  images = pipe_xl(
169
  prompt=prompt,
170
  negative_prompt=negative_prompt,
 
253
  with gr.Row():
254
  model = gr.Radio(
255
  label="Model",
256
+ choices=["Fluently XL v3 Lightning","Fluently XL v4", "Fluently v4", "Fluently Anime", "Fluently Epic", "Fluently XL v3 inpaint", "Fluently v4 inpaint"],
257
  value="Fluently XL v3 Lightning",
258
  interactive=True,
259
  )