Update app.py

app.py

@@ -10,8 +10,6 @@ from diffusers.utils import export_to_video
 import spaces 
 import uuid
 
-is_canonical = True if os.environ.get("SPACE_ID") == "Pyramid-Flow/pyramid-flow" else False
-
 # Constants
 MODEL_PATH = "pyramid-flow-model"
 MODEL_REPO = "rain1011/pyramid-flow-sd3"
@@ -62,7 +60,7 @@ model = load_model()
 
 # Text-to-video generation function
 @spaces.GPU(duration=120)
-def generate_video(prompt, image=None, duration=10, guidance_scale=9, video_guidance_scale=5, progress=gr.Progress(track_tqdm=True)):
+def generate_video(prompt, image=None, duration=5, guidance_scale=9, video_guidance_scale=5, progress=gr.Progress(track_tqdm=True)):
     multiplier = 3
     temp = int(duration * multiplier) + 1  # Convert seconds to temp value (assuming 24 FPS)
     torch_dtype = torch.bfloat16 if MODEL_DTYPE == "bf16" else torch.float32
@@ -109,26 +107,16 @@ 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=5, value=5, step=1, label="Duration (seconds)", visible=not is_canonical)
+                t2v_duration = gr.Slider(minimum=1, maximum=10, value=5, step=1, label="Duration (seconds)")
                 t2v_guidance_scale = gr.Slider(minimum=1, maximum=15, value=7, 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/Pyramid-Flow/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>
-                """)
     gr.Examples(
         examples=[
-            "A movie trailer featuring the adventures of the 30 year old space man wearing a red wool knitted motorcycle helmet, blue sky, salt desert, cinematic style, shot on 35mm film, vivid colors",
-            "Beautiful, snowy Tokyo city is bustling. The camera moves through the bustling city street, following several people enjoying the beautiful snowy weather and shopping at nearby stalls. Gorgeous sakura petals are flying through the wind along with snowflakes"
+            "A futuristic explorer, 30 years old, travels across distant galaxies in a sleek silver space suit, gliding through a glowing nebula. The scene is illuminated by vibrant starbursts and cosmic dust, captured with a futuristic drone in ultra-high-definition, showcasing vibrant purples and blues",
+            "In a serene winter landscape, a futuristic metropolis hums with life. The camera glides along an icy street as citizens, wrapped in advanced thermal suits, enjoy the wintry scene. Holographic advertisements flicker above snow-covered buildings, while sleek flying vehicles zip overhead. In the background, delicate crystalline structures refract light through the snowflakes."
         ],
         fn=generate_video,
         inputs=t2v_prompt,