Update app.py

app.py

@@ -7,6 +7,7 @@ import random
 import spaces
 import torch
 from huggingface_hub import hf_hub_download
+from safetensors.torch import load_file as load_sft
 
 from diffusers import  DiffusionPipeline, FlowMatchEulerDiscreteScheduler, AutoencoderTiny, AutoencoderKL, FluxPipeline
 from transformers import CLIPTextModel, CLIPTokenizer, T5EncoderModel, T5TokenizerFast