Update app.py

app.py

@@ -80,7 +80,7 @@ base_model = 'briaai/BRIA-4B-Adapt'
 controlnet_model = 'briaai/BRIA-4B-Adapt-ControlNet-Union'
 controlnet = BriaControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.bfloat16)
 pipe = BriaControlNetPipeline.from_pretrained(base_model, controlnet=controlnet, torch_dtype=torch.bfloat16, trust_remote_code=True)
-pipe = pipeline.to(device="cuda", dtype=torch.bfloat16)
+pipe = pipe.to(device="cuda", dtype=torch.bfloat16)
 
 mode_mapping = {
     "depth": 0,