change from diffusion pipeline to flux control

app.py

@@ -3,7 +3,8 @@ import numpy as np
 import random
 
 import spaces #[uncomment to use ZeroGPU]
-from diffusers import DiffusionPipeline
+# from diffusers import DiffusionPipeline
+from diffusers import FluxControlPipeline
 from controlnet_aux import CannyDetector
 from huggingface_hub import login
 import torch
@@ -32,7 +33,7 @@ if torch.cuda.is_available():
 else:
     torch_dtype = torch.float32
 
-pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
+pipe = FluxControlPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
 pipe = pipe.to(device)
 pipe.load_lora_weights(flat_lora_path, weight_name=flat_weigths_file, adapter_name="flat")
 pipe.load_lora_weights(canny_lora_path, weight_name=canny_weigths_file, adapter_name="canny")