Update app.py

app.py

@@ -1,4 +1,7 @@
-import os, torch, random
+import spaces
+import os
+import torch
+import random
 from huggingface_hub import snapshot_download
 from kolors.pipelines.pipeline_stable_diffusion_xl_chatglm_256 import StableDiffusionXLPipeline
 from kolors.models.modeling_chatglm import ChatGLMModel
@@ -29,6 +32,7 @@ pipe = StableDiffusionXLPipeline(
 pipe = pipe.to("cuda")
 pipe.enable_model_cpu_offload()
 
+@spaces.GPU
 def generate_image(prompt, height, width, num_inference_steps, guidance_scale):
     seed = random.randint(0, 18446744073709551615)
     image = pipe(