Update app.py

app.py

@@ -7,12 +7,12 @@ from PIL import Image, ImageOps
 vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
 pipeline = AutoPipelineForInpainting.from_pretrained("diffusers/stable-diffusion-xl-1.0-inpainting-0.1", vae=vae, torch_dtype=torch.float16, variant="fp16", use_safetensors=True).to("cuda")
 
-#@spaces.GPU()
+@spaces.GPU()
 def get_select_index(evt: gr.SelectData):
     print(evt)
     return evt.index
     
-#@spaces.GPU()
+@spaces.GPU()
 def squarify_image(img):
     if(img.height > img.width): bg_size = img.height
     else:  bg_size = img.width
@@ -21,7 +21,7 @@ def squarify_image(img):
 
     return bg
 
-#@spaces.GPU()
+@spaces.GPU()
 def divisible_by_8(image):
     width, height = image.size
     
@@ -34,13 +34,13 @@ def divisible_by_8(image):
     
     return resized_image
 
-#@spaces.GPU()
+@spaces.GPU()
 def restore_version(index, versions):
     print('restore version:', index)
     final_dict = {'background': versions[index][0], 'layers': None, 'composite': versions[index][0]}
     return final_dict
 
-#@spaces.GPU()
+@spaces.GPU()
 def generate(image_editor, prompt, neg_prompt, versions):
     image = image_editor['background'].convert('RGB')