Update app.py

app.py

@@ -3,10 +3,9 @@ import gradio as gr
 import numpy as np
 import random
 import os
-from live_preview_helpers import calculate_shift, retrieve_timesteps, flux_pipe_call_that_returns_an_iterable_of_images
-
-DiffusionPipeline, FlowMatchEulerDiscreteScheduler, AutoencoderTiny, AutoencoderKL
 import torch
+from diffusers import DiffusionPipeline, FlowMatchEulerDiscreteScheduler, AutoencoderTiny, AutoencoderKL
+from live_preview_helpers import calculate_shift, retrieve_timesteps, flux_pipe_call_that_returns_an_iterable_of_images
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 hf_token = os.getenv('HF_TOKEN')