Update app.py

app.py

@@ -123,6 +123,7 @@ refiner = StableDiffusionXLImg2ImgPipeline.from_pretrained(
 refiner.watermark = NoWatermark()
 print("Finish Loading Model!")
 
+@spaces.GPU()
 def generate_image(prompt, 
                    n_prompt="", 
                    style: str = DEFAULT_STYLE_NAME,