added spaces

check_app.py

@@ -1,17 +1,12 @@
+import spaces
 import torch
 from diffusers import FluxPipeline, StableDiffusion3Pipeline
 from PIL import Image
 from io import BytesIO
 import gradio as gr
 
-# Initialize pipelines
-stable_diffusion_pipe = StableDiffusion3Pipeline.from_pretrained(
-    "stabilityai/stable-diffusion-3.5-large", torch_dtype=torch.bfloat16
-).to("cuda")
 
-flux_pipe = FluxPipeline.from_pretrained(
-    "black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16
-).to("cuda")
+
 
 # Function to generate images with progress
 def generate_image_with_progress(pipe, prompt, num_steps, guidance_scale=None, seed=None, progress=gr.Progress()):
@@ -44,21 +39,31 @@ def generate_image_with_progress(pipe, prompt, num_steps, guidance_scale=None, s
 
 # Gradio application
 def main():
+    
+    @spaces.GPU(duration=170)
     def tab1_logic(prompt_text):
         progress = gr.Progress()
         num_steps = 30
         seed = 42
         print(f"Start tab {prompt_text}")
+        flux_pipe = FluxPipeline.from_pretrained(
+        "black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16
+        ).to("cuda")
         image = generate_image_with_progress(
             flux_pipe, prompt_text, num_steps=num_steps, seed=seed, progress=progress
         )
         return f"Seed: {seed}", image
-
+    
+    @spaces.GPU(duration=170)
     def tab2_logic(prompt_text):
         progress = gr.Progress()
         num_steps = 28
         guidance_scale = 3.5
         print(f"Start tab {prompt_text}")
+        # Initialize pipelines
+        stable_diffusion_pipe = StableDiffusion3Pipeline.from_pretrained(
+        "stabilityai/stable-diffusion-3.5-large", torch_dtype=torch.bfloat16
+        ).to("cuda")
         image = generate_image_with_progress(
             stable_diffusion_pipe, prompt_text, num_steps=num_steps, guidance_scale=guidance_scale, progress=progress
         )