Update app.py

app.py

@@ -13,6 +13,8 @@ import uuid
 import subprocess
 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 
+is_canonical = True if os.environ.get("SPACE_ID") == "multimodalart/pyramid-flow" else False
+
 # Constants
 MODEL_PATH = "pyramid-flow-model"
 MODEL_REPO = "rain1011/pyramid-flow-sd3"
@@ -62,8 +64,9 @@ def load_model():
 model = load_model()
 
 # Text-to-video generation function
-@spaces.GPU(duration=240)
+@spaces.GPU(duration=120)
 def generate_video(image, prompt, duration, guidance_scale, video_guidance_scale):
+    multiplier = 0.8 if is_canonical else 2.4
     temp = int(duration * 0.8)  # Convert seconds to temp value (assuming 24 FPS)
     torch_dtype = torch.bfloat16 if MODEL_DTYPE == "bf16" else torch.float32
     if(image):
@@ -135,13 +138,22 @@ with gr.Blocks() as demo:
                 i2v_image = gr.Image(type="pil", label="Input Image")
             t2v_prompt = gr.Textbox(label="Prompt")
             with gr.Accordion("Advanced settings", open=False):
-                t2v_duration = gr.Slider(minimum=1, maximum=10, value=5, step=1, label="Duration (seconds)")
+                t2v_duration = gr.Slider(minimum=1, maximum=10, value=5, step=1, label="Duration (seconds)", visible=not is_canonical)
                 t2v_guidance_scale = gr.Slider(minimum=1, maximum=15, value=9, step=0.1, label="Guidance Scale")
                 t2v_video_guidance_scale = gr.Slider(minimum=1, maximum=15, value=5, step=0.1, label="Video Guidance Scale")
             t2v_generate_btn = gr.Button("Generate Video")
         with gr.Column():
             t2v_output = gr.Video(label="Generated Video")
-    
+            gr.HTML("""
+                <div style="display: flex; flex-direction: column;justify-content: center; align-items: center; text-align: center;">
+                    <p style="display: flex;gap: 6px;">
+                         <a href="https://huggingface.co/spaces/multimodalart/pyramid-flow?duplicate=true">
+                            <img src="https://huggingface.co/datasets/huggingface/badges/resolve/main/duplicate-this-space-lg.svg" alt="Duplicate this Space">
+                        </a>
+                    </p>
+                    <p>to use privately and generate videos up to 10s</p>
+                </div>
+                """)
     t2v_generate_btn.click(
         generate_video,
         inputs=[i2v_image, t2v_prompt, t2v_duration, t2v_guidance_scale, t2v_video_guidance_scale],