Update app.py

app.py

@@ -1,15 +1,21 @@
 from diffusers import StableDiffusionXLPipeline, AutoencoderKL
+import random
+import torch
+import numpy as np 
+import spaces
 
 vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
 pipe = StableDiffusionXLPipeline.from_pretrained(
     "stabilityai/stable-diffusion-xl-base-1.0",
     torch_dtype=torch.float16, variant="fp16", use_safetensors=True,
     vae=vae,
-    add_watermarker=False,
 ).to("cuda")
 
 pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 
+MAX_SEED = np.iinfo(np.int32).max
+
+@spaces.GPU
 def run(prompt="a photo of an astronaut riding a horse on mars", steps=10, seed=20, negative_prompt="", randomize_seed=False):
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)