Update app.py

app.py

@@ -2,7 +2,6 @@ import os
 import torch
 from diffusers import DiffusionPipeline
 import gradio as gr
-from peft import PeftModel
 
 # Load the base model and apply the LoRA weights for super realism
 def load_pipeline():
@@ -10,13 +9,16 @@ def load_pipeline():
     lora_repo = "strangerzonehf/Flux-Super-Realism-LoRA"
     trigger_word = "Super Realism"  # Recommended trigger word
 
+    
+
     pipe = DiffusionPipeline.from_pretrained(
         base_model,
         torch_dtype=torch.bfloat16,
+       
     )
 
-    # Load the LoRA weights into the pipeline using PEFT
-    pipe = PeftModel.from_pretrained(pipe, lora_repo)
+    # Load the LoRA weights into the pipeline
+    pipe.load_lora_weights(lora_repo)
 
     # Use GPU if available
     device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -28,9 +30,11 @@ pipe = load_pipeline()
 
 # Define a function for image generation
 def generate_image(prompt, seed, width, height, guidance_scale, randomize_seed):
+    # If randomize_seed is selected, allow the model to generate a random seed
     if randomize_seed:
         seed = None
 
+    # Ensure the prompt includes realism trigger words if needed
     if "realistic" not in prompt.lower() and "realism" not in prompt.lower():
         prompt += " realistic, realism"
 
@@ -48,7 +52,11 @@ def generate_image(prompt, seed, width, height, guidance_scale, randomize_seed):
 iface = gr.Interface(
     fn=generate_image,
     inputs=[
-        gr.Textbox(lines=2, label="Prompt", placeholder="Enter your prompt..."),
+        gr.Textbox(
+            lines=2, 
+            label="Prompt", 
+            placeholder="Enter your prompt, e.g., 'A tiny astronaut hatching from an egg on the moon, 4k, planet theme'"
+        ),
         gr.Slider(0, 10000, step=1, value=0, label="Seed (0 for random)"),
         gr.Slider(256, 1024, step=64, value=1024, label="Width"),
         gr.Slider(256, 1024, step=64, value=1024, label="Height"),
@@ -56,12 +64,13 @@ iface = gr.Interface(
         gr.Checkbox(value=True, label="Randomize Seed")
     ],
     outputs=gr.Image(type="pil", label="Generated Image"),
-    title="Flux Supers Realism LoRA Demo",
+    title="Flux Super Realism LoRA Demo",
     description=(
         "This demo uses the Flux Super Realism LoRA model for ultra-realistic image generation. "
-        "Use the trigger word 'Super Realism' for better results."
+        "You can use the trigger word 'Super Realism' (recommended) along with other realism-related words "
+        "to guide the generation process."
     ),
 )
 
 if __name__ == "__main__":
-    iface.launch(share=False)
+    iface.launch(share=False)