changes small

app.py

@@ -329,7 +329,11 @@ def generate_image_stack_edits(text_prompt, n_edits, input_image):
     return (list_of_images, first_image) so the UI shows immediately.
     """
     DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
+    
+    # if pipeline is null will initialize it simply. 
     global PIPELINE
+    if PIPELINE is None:
+        status = load_pipeline_single_gpu()
     
     if not input_image or not text_prompt or text_prompt.startswith("Please select"):
         return [], None
@@ -520,7 +524,14 @@ def process_user_upload(uploaded_image, user_prompt, n_edits_val):
     
     return processed_image, generated_list, first_result, slider_update
 
+
+@spaces.GPU
+def gpu_warmup():
+    return load_pipeline_single_gpu()
+
 with gr.Blocks() as demo:
+    # warming up the demo for the first run
+    demo.load(gpu_warmup)    
     gr.Markdown("# Kontinuous Kontext - Continuous Strength Control for Instruction-based Image Editing")
     
     # Add description section