Update app.py

app.py

@@ -54,14 +54,14 @@ import requests
 from ltx_core.components.diffusion_steps import Res2sDiffusionStep
 from ltx_core.components.guiders import MultiModalGuider, MultiModalGuiderParams
 from ltx_core.components.noisers import GaussianNoiser
+from ltx_core.components.protocols import DiffusionStepProtocol
 from ltx_core.components.schedulers import LTX2Scheduler
 from ltx_core.model.audio_vae import encode_audio as vae_encode_audio
 from ltx_core.model.audio_vae import decode_audio as vae_decode_audio
 from ltx_core.model.upsampler import upsample_video
 from ltx_core.model.video_vae import TilingConfig, get_video_chunks_number, decode_video as vae_decode_video
 from ltx_core.quantization import QuantizationPolicy
-from ltx_core.types import Audio, AudioLatentShape, VideoPixelShape
-from ltx_core.tools import VideoLatentShape
+from ltx_core.types import Audio, AudioLatentShape, VideoLatentShape, VideoPixelShape
 from ltx_pipelines.distilled import DistilledPipeline
 from ltx_pipelines.utils import (
     res2s_audio_video_denoising_loop,