Update app.py

app.py

@@ -7,8 +7,7 @@ from PIL import Image
 import os
 from huggingface_hub import hf_hub_download
 import torch
-import diffusers
-from pipeline_flux_rf_inversion import RFInversionFluxPipeline
+import diffusers import DiffusionPipeline
 
 # Constants
 MAX_SEED = np.iinfo(np.int32).max
@@ -16,8 +15,9 @@ MAX_IMAGE_SIZE = 1024
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
 
-pipe = RFInversionFluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", 
-                                               torch_dtype=torch.bfloat16)
+pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev",
+                                         custom_pipeline="pipeline_flux_rf_inversion",
+                                          torch_dtype=torch.bfloat16)
 pipe.to(DEVICE)
 
 def reset_do_inversion():