Upload folder using huggingface_hub

app.py

@@ -41,7 +41,19 @@ else:
 
 # Configuration
 VIDEO_SAMPLE_HZ = 1.0
-MAX_FRAMES = 8 if device == "cuda" else 4
+
+# Set MAX_FRAMES based on VRAM
+if device == "cuda":
+    if gpu_mem >= 20.0:  # For A10G (24GB) or 3090/4090
+        MAX_FRAMES = 32
+    elif gpu_mem >= 12.0:  # For T4 (16GB)
+        MAX_FRAMES = 16
+    else:  # For 8GB or less
+        MAX_FRAMES = 8
+else:
+    MAX_FRAMES = 4
+
+print(f"✓ Configured limit: {MAX_FRAMES} frames based on {gpu_mem:.1f} GB VRAM")
 
 # Global model cache
 _vdpm_model = None
@@ -748,4 +760,17 @@ with gr.Blocks(title="DPM-Splat: 4D Gaussian Splatting", theme=gr.themes.Soft())
     )
 
 if __name__ == "__main__":
+    # Download model on startup
+    if device == "cuda":
+        print("Pre-loading VDPM model...")
+        try:
+            get_vdpm_model()
+            global _vdpm_model
+            _vdpm_model = None  # Free VRAM but keep file cached
+            gc.collect()
+            torch.cuda.empty_cache()
+            print("✓ Model pre-loaded and cached")
+        except Exception as e:
+            print(f"⚠ Failed to pre-load model: {e}")
+
     app.queue().launch(share=True, show_error=True)

requirements.txt

@@ -4,6 +4,8 @@ torch
 torchvision
 numpy==1.26.3
 roma
+plyfile
+imageio
 gradio==5.17.1
 pydantic==2.10.6
 matplotlib