Spaces:

toninio19
/

keysync-demo

Running on Zero

Antoni Bigata commited on Apr 16

Commit

2fb3e22

1 Parent(s): fe23184

requirements

Files changed (1) hide show

app.py CHANGED Viewed

@@ -131,16 +131,16 @@ if torch.cuda.is_available():
         print(f"Warning: Failed to compile wavlm_model: {e}")
 landmarks_extractor = LandmarksExtractor()
-keyframe_model = load_model(
-    config="/vol/paramonos2/projects/antoni/code/Personal/code_prep/keysync/scripts/sampling/configs/keyframe.yaml",
-    ckpt="/vol/paramonos2/projects/antoni/code/Personal/code_prep/keysync/pretrained_models/checkpoints/keyframe_dub.pt",
-)
-interpolation_model = load_model(
-    config="/vol/paramonos2/projects/antoni/code/Personal/code_prep/keysync/scripts/sampling/configs/interpolation.yaml",
-    ckpt="/vol/paramonos2/projects/antoni/code/Personal/code_prep/keysync/pretrained_models/checkpoints/interpolation_dub.pt",
-)
-keyframe_model.en_and_decode_n_samples_a_time = 2
-interpolation_model.en_and_decode_n_samples_a_time = 2
 # Default media paths
 DEFAULT_VIDEO_PATH = os.path.join(

         print(f"Warning: Failed to compile wavlm_model: {e}")
 landmarks_extractor = LandmarksExtractor()
+# keyframe_model = load_model(
+#     config="/vol/paramonos2/projects/antoni/code/Personal/code_prep/keysync/scripts/sampling/configs/keyframe.yaml",
+#     ckpt="/vol/paramonos2/projects/antoni/code/Personal/code_prep/keysync/pretrained_models/checkpoints/keyframe_dub.pt",
+# )
+# interpolation_model = load_model(
+#     config="/vol/paramonos2/projects/antoni/code/Personal/code_prep/keysync/scripts/sampling/configs/interpolation.yaml",
+#     ckpt="/vol/paramonos2/projects/antoni/code/Personal/code_prep/keysync/pretrained_models/checkpoints/interpolation_dub.pt",
+# )
+# keyframe_model.en_and_decode_n_samples_a_time = 2
+# interpolation_model.en_and_decode_n_samples_a_time = 2
 # Default media paths
 DEFAULT_VIDEO_PATH = os.path.join(