Update app.py

app.py

@@ -1,6 +1,11 @@
 import gradio as gr
 import torch
-from diffusers import StableDiffusionXLPipeline, EulerDiscreteScheduler
+from diffusers import (
+    StableDiffusionXLPipeline,
+    EulerDiscreteScheduler,
+    AutoencoderKL,
+    DPMSolverSinglestepScheduler,
+)
 from huggingface_hub import hf_hub_download
 
 # --- Configuration ---
@@ -36,14 +41,14 @@ pipe.load_lora_weights(lora_file_path)
 pipe.to("cuda")
 
 # --- Default Settings from your Recommendations ---
-# These are pulled directly from your "Recomendations.txt". 
-default_positive_prompt = "masterpiece, best quality, ultra-detailed, realistic skin, intricate details, highres" # 
-default_negative_prompt = "low quality, worst quality, blurry, (deformed:1.3), extra fingers, cartoon, 3d, anime, bad anatomy" # 
-default_sampler = "DPM++ 2M Karras" # 
-default_cfg = 6.0 # 
-default_steps = 30 # 
-trigger_word = "emilyh" # 
-lora_tag_main = "<lora:emilyh:0.9>" # 
+# These are pulled directly from your "Recomendations.txt".
+default_positive_prompt = "masterpiece, best quality, ultra-detailed, realistic skin, intricate details, highres"
+default_negative_prompt = "low quality, worst quality, blurry, (deformed:1.3), extra fingers, cartoon, 3d, anime, bad anatomy"
+default_sampler = "DPM++ 2M Karras"
+default_cfg = 6.0
+default_steps = 30
+trigger_word = "emilyh"
+lora_tag_main = "<lora:emilyh:0.9>"
 
 
 # --- Define the Inference Function ---
@@ -51,7 +56,7 @@ def generate_image(prompt, negative_prompt, sampler, steps, cfg, width, height,
     """
     Function to generate an image based on user inputs.
     """
-    # Combine the user prompt with the trigger word and LoRA tag 
+    # Combine the user prompt with the trigger word and LoRA tag
     full_prompt = f"{lora_tag_main}, {trigger_word}, {prompt}"
 
     # Set the scheduler (sampler)
@@ -84,7 +89,7 @@ with gr.Blocks(css="style.css") as demo:
     gr.Markdown("# `emilyh` LoRA Image Generator")
     gr.Markdown(
         "A Gradio interface for the `emilyh` LoRA. "
-        "Based on the recommendations provided. "
+        "Based on the recommendations provided."
     )
 
     with gr.Row():
@@ -97,10 +102,10 @@ with gr.Blocks(css="style.css") as demo:
                     label="Sampler",
                     choices=["DPM++ 2M Karras", "DPM++ SDE Karras"],
                     value=default_sampler,
-                ) # 
-                steps = gr.Slider(label="Steps", minimum=15, maximum=50, value=default_steps, step=1) # 
+                )
+                steps = gr.Slider(label="Steps", minimum=15, maximum=50, value=default_steps, step=1)
             
-            cfg = gr.Slider(label="CFG Scale", minimum=1.0, maximum=10.0, value=default_cfg, step=0.5) # 
+            cfg = gr.Slider(label="CFG Scale", minimum=1.0, maximum=10.0, value=default_cfg, step=0.5)
 
             with gr.Row():
                 width = gr.Slider(label="Width", minimum=512, maximum=1024, value=1024, step=64)
@@ -114,12 +119,12 @@ with gr.Blocks(css="style.css") as demo:
             output_image = gr.Image(label="Generated Image", type="pil")
             gr.Markdown(
                 """
-                ### 🔧 Usage Guide 
-                * The trigger word `emilyh` and the LoRA tag `<lora:emilyh:0.9>` are automatically added to your prompt. 
-                * For best results, generate images in batches and choose the most consistent ones. 
-                * The LoRA captures the subject's appearance well across various poses and outfits. 
-                * A weight of 0.9 provides a good balance of likeness and flexibility.  Using a weight closer to 1.0 can increase consistency but may cause stiffness. 
-                * This interface does not include ADetailer, which is recommended for final face refinement. 
+                ### 🔧 Usage Guide
+                * The trigger word `emilyh` and the LoRA tag `<lora:emilyh:0.9>` are automatically added to your prompt.
+                * For best results, generate images in batches and choose the most consistent ones.
+                * The LoRA captures the subject's appearance well across various poses and outfits.
+                * A weight of 0.9 provides a good balance of likeness and flexibility. Using a weight closer to 1.0 can increase consistency but may cause stiffness.
+                * This interface does not include ADetailer, which is recommended for final face refinement.
                 """
             )