added spaces

check_app.py

@@ -1,3 +1,4 @@
+import spaces
 import torch
 from diffusers import (
     FluxPipeline,
@@ -87,9 +88,9 @@ def generate_image_with_progress(pipe, prompt, num_steps, guidance_scale=None, s
 
     return image
 
-
+@spaces.GPU(duration=170)
 def create_pipeline_logic(model_name, config):
-    def logic(prompt_text):
+    def start_process(prompt_text):
         print(f"starting {model_name}")
         progress = gr.Progress()
         num_steps = 30
@@ -123,7 +124,7 @@ def main():
                 img = gr.Image(label=model_name, height=300)
 
                 logic = create_pipeline_logic(model_name, config)
-                button.click(fn=logic, inputs=[prompt_text], outputs=[output, img])
+                button.click(fn=start_process, inputs=[prompt_text], outputs=[output, img])
 
     app.launch()