Update app.py

app.py

@@ -18,9 +18,7 @@ from PIL import Image
 # from featup.util import norm
 from torchaudio.functional import resample
 
-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
+
 from os.path import join
 
 
@@ -82,6 +80,10 @@ models = {o: LitAVAligner.from_pretrained(f"mhamilton723/DenseAV-{o}") for o in
 
 
 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
+    
     # model = models[model_option].cuda()
     model = models[model_option]