Update app.py

app.py

@@ -10,6 +10,7 @@ import time
 from huggingface_hub import hf_hub_download
 from diffusers import FluxTransformer2DModel, FluxPipeline
 import safetensors.torch
+from safetensors.torch import load_file
 import gc
 
 cache_path = path.join(path.dirname(path.abspath(__file__)), "models")