Update app_lora1.py

app_lora1.py

@@ -5,6 +5,7 @@ import traceback
 from diffusers import ZImagePipeline
 from huggingface_hub import list_repo_files
 from PIL import Image
+import spaces
 
 # ============================================================
 # CONFIG
@@ -68,7 +69,7 @@ def list_loras_from_repo(repo_id: str):
 # ============================================================
 # IMAGE GENERATION (SAFE LORA LOGIC)
 # ============================================================
-
+@spaces.GPU()
 def generate_image(prompt, height, width, steps, seed, guidance_scale):
     LOGS = []