Update app.py

app.py

@@ -5,15 +5,11 @@ import torch
 import spaces
 from PIL import Image
 import os
-from pipeline_flux_ipa import FluxPipeline
-from transformer_flux import FluxTransformer2DModel
-from attention_processor import IPAFluxAttnProcessor2_0
 
 from models.transformer_sd3 import SD3Transformer2DModel
 from pipeline_stable_diffusion_3_ipa import StableDiffusion3Pipeline
 
 from transformers import AutoProcessor, SiglipVisionModel
-from infer_flux_ipa_siglip import MLPProjModel, IPAdapter
 from huggingface_hub import hf_hub_download
 
 
@@ -32,7 +28,7 @@ transformer = SD3Transformer2DModel.from_pretrained(
     torch_dtype=torch.bfloat16
 )
 
-pipe = FluxPipeline.from_pretrained(
+pipe = StableDiffusion3Pipeline.from_pretrained(
     model_path, 
     transformer=transformer, 
     torch_dtype=torch.bfloat16