Update app.py

app.py

@@ -6,7 +6,7 @@ from diffusers.pipelines.auto_pipeline import AutoPipelineForImage2Image
 from src.sdxl_inversion_pipeline import SDXLDDIMPipeline
 from src.config import RunConfig
 from src.editor import ImageEditorDemo
-
+import spaces
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 scheduler_class = MyEulerAncestralDiscreteScheduler
@@ -27,7 +27,7 @@ pipe_inversion.scheduler_inference = scheduler_class.from_config(pipe_inference.
 #     pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo", use_safetensors=True)
 #     pipe = pipe.to(device)
 
-
+@spaces.GPU
 def infer(input_image, description_prompt, target_prompt, edit_guidance_scale, num_inference_steps=4,
           num_inversion_steps=4,
           inversion_max_step=0.6):