Update app.py

app.py

@@ -3,7 +3,7 @@ import cv2
 from PIL import Image
 import numpy as np
 
-+import spaces
+import spaces
 from diffusers import StableDiffusionControlNetPipeline, ControlNetModel, UniPCMultistepScheduler
 from diffusers.utils import load_image
 import torch
@@ -14,7 +14,7 @@ from random import randrange
 
 transformers.utils.move_cache()
 
-device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 base_model_id = "runwayml/stable-diffusion-v1-5"
 model_id = "LuyangZ/FloorAI"
@@ -86,7 +86,7 @@ def clean_img(image, mask):
     image = Image.fromarray(image).convert('RGB')
     return image
 
-+@spaces.GPU
+@spaces.GPU
 def floorplan_generation(outline, num_of_rooms):
     new_width = 512
     new_height = 512