Update app.py

app.py

@@ -14,6 +14,7 @@
 import os
 import torch
 import mediapy
+import data
 import subprocess
 from einops import rearrange
 from omegaconf import OmegaConf