Update

app.py

@@ -6,6 +6,7 @@ from PIL import Image, ImageDraw
 from diffusers import DiffusionPipeline
 import gradio as gr
 from gradio.components.image_editor import EditorValue
+import spaces
 
 DEVICE = "cuda"
 
@@ -35,7 +36,7 @@ def crop_divisible_by_16(image: Image.Image) -> Image.Image:
     return image.crop((0, 0, w, h))
 
 
-# @spaces.GPU(duration=150)
+@spaces.GPU(duration=150)
 def predict(
     image_and_mask: EditorValue,
     condition_image: Image.Image | None,