add spaces

app.py

@@ -24,6 +24,7 @@ import subprocess
 from moviepy.editor import VideoFileClip
 from huggingface_hub import snapshot_download
 import shutil
+import spaces
 try:
     from audio_separator.separator import Separator
 except:
@@ -150,7 +151,7 @@ pipeline = WanI2VTalkingInferenceLongPipeline(
     wav2vec=wav2vec,
 )
 
-
+@spaces.GPU(duration=6000)
 def generate(
     GPU_memory_mode,
     teacache_threshold,
@@ -421,13 +422,13 @@ BANNER_HTML = """
 
 BANNER_CSS = """
 .hero{display:flex;align-items:center;gap:18px;padding:18px;border-radius:14px;
-      background:#111;color:#fff;margin-bottom:12px}
+      color:inherit;margin-bottom:12px}
 .brand-text{font-weight:800;letter-spacing:2px}
 .brand img{height:46px}
 .titles h1{font-size:28px;margin:0 0 6px 0}
 .badges{display:flex;gap:10px;flex-wrap:wrap}
 .badge img{height:22px}
-.divider{border:0;border-top:1px solid rgba(255,255,255,0.18);margin:6px 0 18px}
+.divider{border:0;border-top:1px solid rgba(0,0,0,0.12);margin:6px 0 18px}
 """
 
 

requirements.txt

@@ -4,12 +4,10 @@ einops
 safetensors
 timm
 tomesd
-torch==2.7.0
 torchdiffeq
 torchsde
 decord
 datasets
-torchvision==0.22.0
 opencv-python>=4.9.0.80
 diffusers==0.30.1
 transformers==4.51.3
@@ -23,7 +21,6 @@ imageio-ffmpeg
 gradio>=5.0.0
 numpy>=1.23.5,<2
 scikit-image
-opencv-python
 omegaconf
 SentencePiece
 albumentations
@@ -33,5 +30,4 @@ tensorboard
 beautifulsoup4
 ftfy
 librosa
-torchaudio==2.7.0
 moviepy==1.0.3