Update app.py

app.py

@@ -10,6 +10,18 @@ from diffusers.utils import load_image
 from skyreels_v2_infer.modules import download_model
 from skyreels_v2_infer.pipelines import Image2VideoPipeline, Text2VideoPipeline, PromptEnhancer, resizecrop
 
+MODEL_ID_CONFIG = {
+    "text2video": [
+        "Skywork/SkyReels-V2-T2V-14B-540P",
+        "Skywork/SkyReels-V2-T2V-14B-720P",
+    ],
+    "image2video": [
+        "Skywork/SkyReels-V2-I2V-1.3B-540P",
+        "Skywork/SkyReels-V2-I2V-14B-540P",
+        "Skywork/SkyReels-V2-I2V-14B-720P",
+    ],
+}
+
 def generate_video(
     prompt,
     model_id,