Update app.py

app.py

@@ -282,11 +282,11 @@ def generate_video(uploaded_img, uploaded_audio, infer_type,
         return None, gr.Markdown("Error: Input image or audio file is empty. Please check and upload both files.")
     
     model_mapping = {
-        "mfcc_pose_only": "./ckpt/stage2_pose_only_mfcc.ckpt",
-        "mfcc_full_control": "./ckpt/stage2_more_controllable_mfcc.ckpt",
-        "hubert_audio_only": "./ckpt/stage2_audio_only_hubert.ckpt",
-        "hubert_pose_only": "./ckpt/stage2_pose_only_hubert.ckpt",
-        "hubert_full_control": "./ckpt/stage2_full_control_hubert.ckpt",
+        "mfcc_pose_only": "ckpt/stage2_pose_only_mfcc.ckpt",
+        "mfcc_full_control": "ckpt/stage2_more_controllable_mfcc.ckpt",
+        "hubert_audio_only": "ckpt/stage2_audio_only_hubert.ckpt",
+        "hubert_pose_only": "ckpt/stage2_pose_only_hubert.ckpt",
+        "hubert_full_control": "ckpt/stage2_full_control_hubert.ckpt",
     }
 
     # if face_crop:
@@ -305,7 +305,7 @@ def generate_video(uploaded_img, uploaded_audio, infer_type,
             test_audio_path=uploaded_audio,
             test_hubert_path='',
             result_path='./outputs/',
-            stage1_checkpoint_path='./ckpt/stage1.ckpt',
+            stage1_checkpoint_path='ckpt/stage1.ckpt',
             stage2_checkpoint_path=stage2_checkpoint_path,
             seed=seed,
             control_flag=True,