requirements

app.py

@@ -167,9 +167,24 @@ def load_all_models():
     )
     keyframe_model.en_and_decode_n_samples_a_time = 2
     interpolation_model.en_and_decode_n_samples_a_time = 2
+    return (
+        keyframe_model,
+        interpolation_model,
+        vae_model,
+        hubert_model,
+        wavlm_model,
+        landmarks_extractor,
+    )
 
 
-load_all_models()
+(
+    keyframe_model,
+    interpolation_model,
+    vae_model,
+    hubert_model,
+    wavlm_model,
+    landmarks_extractor,
+) = load_all_models()
 
 
 @spaces.GPU(duration=60)