Update app.py

app.py

@@ -11,11 +11,10 @@ from __future__ import annotations
 import spaces
 import gradio as gr
 import numpy as np
-from PIL import Image, ImageFilter, ImageEnhance
+from PIL import Image
 import tempfile
 import os
 import random
-import time
 from typing import Optional
 import imageio
 
@@ -154,7 +153,7 @@ def generate_ken_burns_video(
     num_frames: int,
     motion_type: str,
     strength: float,
-) -> list[np.ndarray]:
+) -> list:
     """Create a Ken Burns style animation from a still image."""
     img = image.convert("RGB")
 
@@ -471,6 +470,7 @@ def generate_ad_headlines(preset: str, seed: int) -> str:
 # Main generation pipeline
 # ---------------------------------------------------------------------------
 
+@spaces.GPU
 def generate_video(
     image,
     preset: str,
@@ -478,7 +478,7 @@ def generate_video(
     video_length: str,
     aspect_ratio: str,
     motion_strength: str,
-    seed_value: int,
+    seed_value,
     progress=gr.Progress(),
 ):
     """Main generation function called by the Gradio interface."""
@@ -490,7 +490,8 @@ def generate_video(
 
     # Parse inputs
     duration = int(video_length.split()[0])
-    seed = seed_value if seed_value and seed_value > 0 else random.randint(1, 999999)
+    seed_val = int(seed_value) if seed_value else 0
+    seed = seed_val if seed_val > 0 else random.randint(1, 999999)
 
     # Build the prompt
     base_prompt = PRESET_PROMPTS.get(preset, "")
@@ -581,12 +582,6 @@ CUSTOM_CSS = """
     margin-top: 0;
     margin-bottom: 20px;
 }
-.output-section {
-    border: 1px solid #e0e0e0;
-    border-radius: 8px;
-    padding: 15px;
-    margin-top: 10px;
-}
 footer {
     text-align: center;
     padding: 20px;