Update app.py

app.py

@@ -14,6 +14,8 @@ pipe = CogVideoXPipeline.from_pretrained(
 )
 pipe.enable_model_cpu_offload()   # auto move submodules between CPU/GPU
 pipe.vae.enable_slicing()         # slice VAE for extra VRAM savings
+pipe.enable_attention_slicing()    
+pipe.enable_unet_slicing()  
 
 # ────────────────────────────────────────────────────────────
 # 2. Resolution parsing & sanitization
@@ -60,8 +62,15 @@ def generate_video(
 
     # 3.4 Export to MP4 (H.264) with chosen FPS
     video_path = export_to_video(resized_frames, "generated.mp4", fps=fps)
+
+    import torch
+    torch.cuda.empty_cache()
+    pipe.to("cpu")
+    
     return video_path
 
+    
+
 # ────────────────────────────────────────────────────────────
 # 4. Build the Gradio interface with interactive controls
 # ────────────────────────────────────────────────────────────