Update app.py

app.py

@@ -16,7 +16,7 @@ from random import randrange
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
 base_model_id = "runwayml/stable-diffusion-v1-5"
-model_id = "LuyangZ/controlnet_Neufert4_64_100"
+model_id = "LuyangZ/FloorAI"
 
 controlnet = ControlNetModel.from_pretrained(model_id, torch_dtype=torch.float32)
 controlnet.to(device)
@@ -92,7 +92,7 @@ def floorplan_generation(outline, num_of_rooms):
     n_outline = n_outline.resize((new_width, new_height))
 
     num_of_rooms = str(num_of_rooms)
-    validation_prompt = "floor plan," + num_of_rooms + " bedrooms"
+    validation_prompt = "floor plan, " + num_of_rooms + " rooms"
     validation_image = n_outline
 
     image = pipeline(validation_prompt,