Update app.py

app.py

@@ -14,6 +14,9 @@ SAVE_DIR = "saved_images"  # Gradio will handle the persistence
 if not os.path.exists(SAVE_DIR):
     os.makedirs(SAVE_DIR, exist_ok=True)
 
+# Load the default image
+DEFAULT_IMAGE_PATH = "cover1.webp"
+
 device = "cuda" if torch.cuda.is_available() else "cpu"
 repo_id = "black-forest-labs/FLUX.1-dev"
 adapter_id = "prithivMLmods/EBook-Creative-Cover-Flux-LoRA"
@@ -129,7 +132,12 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css, analytics_enabled=False) as demo
                     )
                     run_button = gr.Button("Run", scale=0)
 
-                result = gr.Image(label="Result", show_label=False)
+                # Modified to include the default image
+                result = gr.Image(
+                    label="Result",
+                    show_label=False,
+                    value=DEFAULT_IMAGE_PATH  # Set the default image
+                )
 
                 with gr.Accordion("Advanced Settings", open=False):
                     seed = gr.Slider(
@@ -198,7 +206,6 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css, analytics_enabled=False) as demo
             )
             refresh_btn = gr.Button("🔄 Refresh Gallery")
 
-
     # Event handlers
     def refresh_gallery():
         return load_generated_images()