Update app_df.py

app_df.py

@@ -11,6 +11,8 @@ from skyreels_v2_infer import DiffusionForcingPipeline
 from skyreels_v2_infer.modules import download_model
 from skyreels_v2_infer.pipelines import PromptEnhancer, resizecrop
 
+is_shared_ui = True if "fffiloni/SkyReels-V2" in os.environ['SPACE_ID'] else False
+
 def generate_diffusion_forced_video(
     prompt,
     model_id,
@@ -33,7 +35,8 @@ def generate_diffusion_forced_video(
     prompt_enhancer=False,
     teacache=False,
     teacache_thresh=0.2,
-    use_ret_steps=False
+    use_ret_steps=False,
+    progress=gr.Progress(track_tqdm=True)
 ):
     model_id = download_model(model_id)
 
@@ -60,7 +63,7 @@ def generate_diffusion_forced_video(
             height, width = width, height
         image = resizecrop(image, height, width)
 
-    negative_prompt = "色调艳丽，过曝，静态，细节模糊不清，字幕，风格，作品，画作，画面，静止，整体发灰，最差质量，低质量，JPEG压缩残留，丑陋的，残缺的，多余的手指，画得不好的手部，画得不好的脸部，畸形的，毁容的，形态畸形的肢体，手指融合，静止不动的画面，杂乱的背景，三条腿，背景人很多，倒着走"
+    negative_prompt = "Bright tones, overexposed, static, blurred details, subtitles, style, works, paintings, images, static, overall gray, worst quality, low quality, JPEG compression residue, ugly, incomplete, extra fingers, poorly drawn hands, poorly drawn faces, deformed, disfigured, misshapen limbs, fused fingers, still picture, messy background, three legs, many people in the background, walking backwards"
 
     prompt_input = prompt
     if prompt_enhancer and image is None:
@@ -130,9 +133,22 @@ model_options = ["Skywork/SkyReels-V2-DF-1.3B-540P"]  # Update if there are more
 
 with gr.Blocks() as demo:
     with gr.Column():
-        gr.Markdown("# SkyReels V2")
-        gr.Markdown("Infinite-Length Film Generation")
+        gr.Markdown("# SkyReels V2: Infinite-Length Film Generation")
+        gr.Markdown("The first open-source video generative model employing AutoRegressive Diffusion-Forcing architecture that achieves the SOTA performance among publicly available models.")
 
+        gr.HTML("""
+            <div style="display:flex;column-gap:4px;">
+                <a href="https://github.com/SkyworkAI/SkyReels-V2">
+                    <img src='https://img.shields.io/badge/GitHub-Repo-blue'>
+                </a> 
+    			<a href="https://arxiv.org/pdf/2504.13074">
+                    <img src='https://img.shields.io/badge/ArXiv-Paper-red'>
+                </a>
+                <a href="https://huggingface.co/spaces/fffiloni/SkyReels-V2?duplicate=true">
+    				<img src="https://huggingface.co/datasets/huggingface/badges/resolve/main/duplicate-this-space-sm.svg" alt="Duplicate this Space">
+    			</a>	
+            </div>
+        """)
         with gr.Row():
 
             with gr.Column():
@@ -185,25 +201,35 @@ with gr.Blocks() as demo:
     def set_num_frames(target_l):
 
         n_frames = 0
+        overlap_history = 0
+        addnoise_condition = 0
 
         if target_l == "4":
             n_frames = 97
         elif target_l == "10":
             n_frames = 257
+            overlap_history = 17
+            addnoise_condition = 20
         elif target_l == "15":
             n_frames = 377
+            overlap_history = 17
+            addnoise_condition = 20
         elif target_l == "30":
             n_frames = 737
+            overlap_history = 17
+            addnoise_condition = 20
         elif target_l == "60":
             n_frames = 1457
+            overlap_history = 17
+            addnoise_condition = 20
         
-        return n_frames
+        return n_frames, overlap_history, addnoise_condition
         
 
     target_length.change(
         fn = set_num_frames,
         inputs = [target_length],
-        outputs = [num_frames],
+        outputs = [num_frames, overlap_history, addnoise_condition],
         queue = False
     )