Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import os
 os.system("pip install diffsynth==2.0.3")
 from modelscope.hub.api import HubApi
@@ -40,6 +41,7 @@ pipe = ZImagePipeline.from_pretrained(
     tokenizer_config=ModelConfig(model_id="Tongyi-MAI/Z-Image-Turbo", origin_file_pattern="tokenizer/"),
 )
 
+@space.GPU(duration=60)
 def run_inference(style_image_1, style_image_2, style_image_3, style_image_4, style_image_5, style_image_6, prompt, negative_prompt, cfg_scale, sigma_shift, seed, num_inference_steps, height, width, progress=gr.Progress()):
     # Filter out None values and collect valid images
     style_images = [img for img in [style_image_1, style_image_2, style_image_3, style_image_4, style_image_5, style_image_6] if img is not None]