Update app.py

app.py

@@ -17,7 +17,8 @@ from random import randrange
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 base_model_id = "runwayml/stable-diffusion-v1-5"
-model_id = "LuyangZ/FloorAI"
+# model_id = "LuyangZ/FloorAI"
+model_id = "LuyangZ/controlnet_Neufert4_64_100"
 
 # controlnet = ControlNetModel.from_pretrained(model_id, torch_dtype=torch.float16)
 # controlnet = ControlNetModel.from_pretrained(model_id, torch_dtype="auto")