Update app.py

app.py

@@ -28,13 +28,13 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
 
 if torch.cuda.is_available():
-    pipe_3_5 = StableDiffusionPipeline.from_pretrained(
-        "fluently/Fluently-v3.5",
+    pipe_4 = StableDiffusionPipeline.from_pretrained(
+        "fluently/Fluently-v4",
         torch_dtype=torch.float16,
         use_safetensors=True,
     )
-    pipe_3_5.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe_3_5.scheduler.config)
-    pipe_3_5.to(device)   
+    pipe_4.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe_4.scheduler.config)
+    pipe_4.to(device)   
 
     pipe_anime = StableDiffusionPipeline.from_pretrained(
         "fluently/Fluently-anime",
@@ -53,7 +53,7 @@ if torch.cuda.is_available():
     pipe_epic.to(device)
 
     pipe_inpaint = StableDiffusionInpaintPipeline.from_pretrained(
-        "fluently/Fluently-v3-inpainting",
+        "fluently/Fluently-v4-inpainting",
         torch_dtype=torch.float16,
         use_safetensors=True,
     )
@@ -84,7 +84,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     return seed
 
 def get_model(model):
-    if model == "Fluently v3 inpaint":
+    if model == "Fluently v4 inpaint":
         return gr.update(visible=True), gr.update(visible=True), gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
     else:
         return gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
@@ -113,8 +113,8 @@ def generate(
 
     if not use_negative_prompt:
         negative_prompt = ""  # type: ignore
-    if model == "Fluently v3.5":
-        images = pipe_3_5(
+    if model == "Fluently v4":
+        images = pipe_4(
             prompt=prompt,
             negative_prompt=negative_prompt,
             width=width,
@@ -209,8 +209,8 @@ with gr.Blocks(title="Fluently Playground", css=css) as demo:
     with gr.Row():
         model = gr.Radio(
             label="Model",
-            choices=["Fluently XL v2", "Fluently v3.5", "Fluently Anime", "Fluently Epic", "Fluently v3 inpaint"],
-            value="Fluently v3.5",
+            choices=["Fluently XL v2", "Fluently v4", "Fluently Anime", "Fluently Epic", "Fluently v4 inpaint"],
+            value="Fluently v4",
             interactive=True,
         )
 
@@ -239,6 +239,7 @@ with gr.Blocks(title="Fluently Playground", css=css) as demo:
         negative_prompt = gr.Text(
             label="Negative prompt",
             max_lines=5,
+            lines=4,
             value="""(deformed, distorted, disfigured:1.3), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers:1.4), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation""",
             placeholder="Enter a negative prompt",
             visible=False,