Update app.py

app.py

@@ -175,6 +175,39 @@ pipeline = TI2VidTwoStagesHQPipeline(
 print("Pipeline initialized successfully!")
 print("=" * 80)
 
+# =============================================================================
+# ZeroGPU Tensor Preloading
+# =============================================================================
+# ZeroGPU needs all tensors to be loaded BEFORE the Space starts.
+# We trigger model loading here so ZeroGPU can pack them into shared GPU memory.
+
+print("Preloading all models for ZeroGPU tensor packing...")
+print("This may take a few minutes...")
+
+# Access all pipeline components to force tensor loading
+# These are accessed but stored as lambda closures so ZeroGPU packs them
+_transformer = pipeline.stage_1.transformer
+_video_encoder = pipeline.prompt_encoder.video_encoder
+_video_decoder = pipeline.video_decoder
+_audio_decoder = pipeline.audio_decoder
+_spatial_upsampler = pipeline.upsampler
+
+# Trigger actual loading by calling them once
+# This forces tensors into memory so ZeroGPU can pack them
+print("  Loading transformer...")
+_transformer()
+print("  Loading video encoder...")
+_video_encoder()
+print("  Loading video decoder...")
+_video_decoder()
+print("  Loading audio decoder...")
+_audio_decoder()
+print("  Loading spatial upsampler...")
+_spatial_upsampler()
+
+print("All models preloaded for ZeroGPU tensor packing!")
+print("=" * 80)
+
 # =============================================================================
 # Helper Functions
 # =============================================================================
@@ -310,10 +343,10 @@ def get_duration(
         base += 15
     
     # More frames means more processing
-    if num_frames > 81:
+    # Calculate num_frames inside get_duration since it's no longer a parameter
+    frames_from_duration = int(duration * DEFAULT_FRAME_RATE)
+    if frames_from_duration > 81:
         base += 10
-    
-    return min(base, 90)
 
 
 @spaces.GPU(duration=get_duration)