Update app.py

app.py

@@ -5,6 +5,7 @@ import time
 import os
 import random
 from skyreelsinfer.offload import OffloadConfig
+from skyreelsinfer import TaskType
 from skyreelsinfer.skyreels_video_infer import SkyReelsVideoInfer
 from diffusers.utils import export_to_video
 from diffusers.utils import load_image