Update app.py

app.py

@@ -3,6 +3,7 @@ import cv2
 from PIL import Image
 import numpy as np
 
++import spaces
 from diffusers import StableDiffusionControlNetPipeline, ControlNetModel, UniPCMultistepScheduler
 from diffusers.utils import load_image
 import torch
@@ -84,7 +85,8 @@ def clean_img(image, mask):
     image[mask<250]=(255,255,255)
     image = Image.fromarray(image).convert('RGB')
     return image
-    
+
+[email protected]
 def floorplan_generation(outline, num_of_rooms):
     new_width = 512
     new_height = 512