Update app.py

app.py

@@ -4,15 +4,23 @@ import spaces
 import torch
 import random
 from PIL import Image
+from huggingface_hub import hf_hub_download
 
 from diffusers import FluxPipeline, FluxTransformer2DModel, GGUFQuantizationConfig
 from diffusers.utils import load_image
 
 MAX_SEED = np.iinfo(np.int32).max
 
-# Load GGUF transformer model from Hugging Face
+# Download GGUF model file first
+gguf_file = hf_hub_download(
+    repo_id="bullerwins/FLUX.1-Kontext-dev-GGUF",
+    filename="flux1-kontext-dev-Q8_0.gguf",
+    local_dir="./models"
+)
+
+# Load GGUF transformer model
 transformer = FluxTransformer2DModel.from_single_file(
-    "https://huggingface.co/bullerwins/FLUX.1-Kontext-dev-GGUF/resolve/main/flux1-kontext-dev-Q8_0.gguf",
+    gguf_file,
     quantization_config=GGUFQuantizationConfig(compute_dtype=torch.bfloat16),
     torch_dtype=torch.bfloat16,
 )