Update app.py

app.py

@@ -6,7 +6,6 @@ from datetime import datetime
 import gradio as gr
 import spaces
 import torch
-from diffusers import FluxPipeline
 from diffusers import FluxControlNetModel
 from diffusers import FluxControlNetPipeline
 from PIL import Image
@@ -16,7 +15,7 @@ from aoti import aoti_load
 
 
 controlnet = FluxControlNetModel.from_pretrained('InstantX/FLUX.1-dev-controlnet-canny', torch_dtype=torch.bfloat16)
-pipeline = FluxPipeline.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.bfloat16, controlnet=controlnet).to('cuda')
+pipeline = FluxControlNetPipeline.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.bfloat16, controlnet=controlnet).to('cuda')
 pipeline.transformer.fuse_qkv_projections()
 aoti_load(pipeline.transformer, 'zerogpu-aoti/FLUX.1')
 
@@ -42,6 +41,5 @@ gr.Interface(
     fn=generate_image,
     inputs=['image', 'text'],
     outputs=['gallery'],
-    examples=["A cat playing with a ball of yarn"],
     cache_examples=False,
 ).launch()