update

app.py

@@ -221,7 +221,7 @@ def generation_step(runner, text_embeds_dict, cond_latents):
     return samples
 
 @spaces.GPU(duration=120)
-def generation_loop(video_path='./test_videos', seed=666, batch_size=1, cfg_scale=1.0, cfg_rescale=0.0, sample_steps=1, res_h=1280, res_w=720, sp_size=1):
+def generation_loop(video_path='./test_videos', seed=666, fps_out=12, batch_size=1, cfg_scale=1.0, cfg_rescale=0.0, sample_steps=1, res_h=1280, res_w=720, sp_size=1):
     runner = configure_runner(1)
     output_dir = 'output/out.mp4'
     def _build_pos_and_neg_prompt():
@@ -386,7 +386,7 @@ def generation_loop(video_path='./test_videos', seed=666, batch_size=1, cfg_scal
             sample = sample.to(torch.uint8).numpy()
 
             mediapy.write_video(
-                output_dir, sample, fps=self.config.dit_generation.get("fps", 12)
+                output_dir, sample, fps=fps_out
             )
 
         print(f"Generated video size: {sample.size()}")
@@ -459,6 +459,7 @@ demo = gr.Interface(
             label="Upload a video"
         ),
         gr.Number(label="Seeds"),
+        gr.Number(label="fps"),
 	], [
 		gr.Video(label="Output"),
 		gr.File(label="Download the output")
@@ -467,9 +468,9 @@ demo = gr.Interface(
 	description=description,
 	article=article,
 	examples=[
-		['./01.mp4', 4],
-		['./02.mp4', 4],
-		['./03.mp4', 4],
+		['./01.mp4', 4, 24],
+		['./02.mp4', 4, 24],
+		['./03.mp4', 4, 24],
 		]
 	)