Update app.py

app.py

@@ -83,6 +83,11 @@ def process_video(video, model_option):
     from denseav.train import LitAVAligner
     from denseav.plotting import plot_attention_video, plot_2head_attention_video, plot_feature_video
     from denseav.shared import norm, crop_to_divisor, blur_dim
+
+    global models
+
+    if len(models) == 0:
+        models = {o: LitAVAligner.from_pretrained(f"mhamilton723/DenseAV-{o}") for o in options}
     
     # model = models[model_option].cuda()
     model = models[model_option]