Update app.py

app.py

@@ -1,9 +1,8 @@
 import gradio as gr
 import torch
 import spaces
-from diffusers import AutoencoderKL, FluxTransformer2DModel, FluxControlNetInpaintPipeline
+from diffusers import AutoencoderKL, FluxTransformer2DModel, FluxControlNetModel, FluxControlNetInpaintPipeline
 from diffusers.utils import load_image
-from controlnet_flux import FluxControlNetModel
 from transformer_flux import FluxTransformer2DModel
 from transformers import T5EncoderModel, CLIPTextModel
 from PIL import Image, ImageDraw