Update app.py

app.py

@@ -8,19 +8,19 @@ import gradio as gr
 from torchvision import transforms 
 
 controlnet = ControlNetModel.from_pretrained(
-    "briaai/ControlNet-Canny",
+    "briaai/BRIA-2.2-ControlNet-Canny",
     torch_dtype=torch.float16
 ).to('cuda')
 
 pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
-    "briaai/BRIA-2.0",
+    "briaai/BRIA-2.2",
     controlnet=controlnet,
     torch_dtype=torch.float16,
     device_map='auto',
     low_cpu_mem_usage=True,
     offload_state_dict=True,
 ).to('cuda')
-pipe.enable_freeu(b1=1.1, b2=1.1, s1=0.5, s2=0.7)
+# pipe.enable_freeu(b1=1.1, b2=1.1, s1=0.5, s2=0.7)
 pipe.enable_xformers_memory_efficient_attention()
 pipe.force_zeros_for_empty_prompt = False