try to load latest diffusers

app.py

@@ -2,7 +2,7 @@ import spaces  # Must be imported first for ZeroGPU
 import gradio as gr
 import torch
 from huggingface_hub import hf_hub_download
-from diffusers import DiffusionPipeline, FluxPipeline, FluxImg2ImgPipeline
+from diffusers import DiffusionPipeline, FluxPipeline, FluxKontextPipeline, FluxImg2ImgPipeline
 from diffusers.utils import load_image
 from PIL import Image
 import os
@@ -50,10 +50,10 @@ def load_pipeline():
         print("Loading FLUX Kontext pipeline...")
         try:
             # Try FluxImg2ImgPipeline first
-            pipeline = DiffusionPipeline.from_pretrained(
+            pipeline = FluxKontextPipeline.from_pretrained(
                 "black-forest-labs/FLUX.1-Kontext-dev", 
                 torch_dtype=dtype,
-                token=os.environ.get("HF_TOKEN"),
+                use_auth_token=os.environ.get("HF_TOKEN"),
             ).to('cuda')
             print("Pipeline loaded successfully with FluxImg2ImgPipeline!")
         except Exception as e:

requirements.txt

@@ -1,5 +1,5 @@
 torch>=2.0.0
-diffusers>=0.34.0
+git+https://github.com/huggingface/diffusers.git@main
 transformers>=4.38.0
 accelerate>=0.26.0
 safetensors>=0.4.0