Update app.py

app.py

@@ -9,9 +9,12 @@ from diffusers import DiffusionPipeline, AutoencoderTiny, AutoencoderKL
 import copy
 import random
 import time
+from diffusers.models.transformers import FluxTransformer2DModel
+import safetensors.torch
+from transformers import CLIPModel, CLIPProcessor, CLIPTextModel, CLIPTokenizer, CLIPConfig, T5EncoderModel, T5Tokenizer
 from live_preview_helpers import calculate_shift, retrieve_timesteps, flux_pipe_call_that_returns_an_iterable_of_images
 from huggingface_hub import HfFileSystem, ModelCard
-
+from safetensors.torch import load_file
 from huggingface_hub import login
 hf_token = os.environ.get("HF_TOKEN")
 login(token=hf_token)