Update app.py

app.py

@@ -9,7 +9,13 @@ from safetensors.torch import load_file
 from omegaconf import OmegaConf
 
 from image_datasets.dataset import image_resize
-from src.flux.util import load_ae, load_clip, load_flow_model2, load_t5, tensor_to_pil_image
+def tensor_to_pil_image(in_image):
+    tensor = in_image.squeeze(0)
+    tensor = (tensor + 1) / 2
+    tensor = tensor * 255
+    numpy_array = tensor.permute(1, 2, 0).byte().numpy()
+    pil_image = Image.fromarray(numpy_array)
+    return pil_image
 from src.flux.xflux_pipeline import XFluxSampler
 args = OmegaConf.load("inference_configs/inference.yaml")
 # is_schnell = args.model_name == "flux-schnell"