Update app.py

app.py

@@ -15,7 +15,7 @@ from diffusers.utils import load_image
 DESCRIPTION = """
 # [Fluently Playground](https://huggingface.co/fluently)
 
-🤗 New FluentlyXL v4!
+[🦾 New FluentlyXL Final!](https://huggingface.co/fluently/Fluently-XL-Final)
 """
 if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>Running on CPU 🥶 This demo may not work on CPU.</p>"
@@ -29,13 +29,13 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
 
 if torch.cuda.is_available():
-    pipe_4 = StableDiffusionPipeline.from_pretrained(
-        "fluently/Fluently-v4",
+    pipe_xl_final = StableDiffusionXLPipeline.from_pretrained(
+        "fluently/Fluently-XL-Final",
         torch_dtype=torch.float16,
         use_safetensors=True,
     )
-    pipe_4.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe_4.scheduler.config)
-    pipe_4.to(device)   
+    pipe_xl_final.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe_xl_final.scheduler.config)
+    pipe_xl_final.to(device)   
 
     pipe_anime = StableDiffusionPipeline.from_pretrained(
         "fluently/Fluently-anime",
@@ -131,14 +131,14 @@ def generate(
 
     if not use_negative_prompt:
         negative_prompt = ""  # type: ignore
-    if model == "Fluently v4":
-        images = pipe_4(
+    if model == "Fluently XL Final":
+        images = pipe_xl_final(
             prompt=prompt,
             negative_prompt=negative_prompt,
             width=width,
             height=height,
             guidance_scale=guidance_scale,
-            num_inference_steps=30,
+            num_inference_steps=25,
             num_images_per_prompt=1,
             output_type="pil",
         ).images
@@ -253,7 +253,7 @@ with gr.Blocks(title="Fluently Playground", css=css) as demo:
     with gr.Row():
         model = gr.Radio(
             label="Model",
-            choices=["Fluently XL v3 Lightning","Fluently XL v4", "Fluently v4", "Fluently Anime", "Fluently Epic", "Fluently XL v3 inpaint", "Fluently v4 inpaint"],
+            choices=["Fluently XL Final", "Fluently XL v4", "Fluently XL v3 Lightning", "Fluently Anime", "Fluently Epic", "Fluently XL v3 inpaint", "Fluently v4 inpaint"],
             value="Fluently XL v3 Lightning",
             interactive=True,
         )