Update app.py

app.py

@@ -2,25 +2,21 @@ import os
 import torch
 from diffusers import DiffusionPipeline
 import gradio as gr
-import zerogpu
+from peft import PeftModel
 
-zerogpu.initialize()
 # Load the base model and apply the LoRA weights for super realism
 def load_pipeline():
     base_model = "black-forest-labs/FLUX.1-dev"
     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
-    pipe.load_lora_weights(lora_repo)
+    # Load the LoRA weights into the pipeline using PEFT
+    pipe = PeftModel.from_pretrained(pipe, lora_repo)
 
     # Use GPU if available
     device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -32,11 +28,9 @@ 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"
 
@@ -54,11 +48,7 @@ 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, e.g., 'A tiny astronaut hatching from an egg on the moon, 4k, planet theme'"
-        ),
+        gr.Textbox(lines=2, label="Prompt", placeholder="Enter your prompt..."),
         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"),
@@ -69,8 +59,7 @@ iface = gr.Interface(
     title="Flux Super Realism LoRA Demo",
     description=(
         "This demo uses the Flux Super Realism LoRA model for ultra-realistic image generation. "
-        "You can use the trigger word 'Super Realism' (recommended) along with other realism-related words "
-        "to guide the generation process."
+        "Use the trigger word 'Super Realism' for better results."
     ),
 )