diff --git a/inferno_apps/TalkingHead/README.md b/inferno_apps/TalkingHead/README.md index 113fc110..fac9b8e1 100644 --- a/inferno_apps/TalkingHead/README.md +++ b/inferno_apps/TalkingHead/README.md @@ -38,6 +38,9 @@ This is the official implementation of [EMOTE : Emotional Speech-Driven Animation with Copntent-Emotion Disentanglement](https://emote.is.tue.mpg.de/). EMOTE takes speech audio and an emotion and intensity labels on the input and produces a talking head avatar that correctly articulates the words spoken in the audio while expressing the specified emotion. +## News +- (13th Dec. 2023) EMOTE v2 is now out! It is trained on a newer iteration of the data and should give better results overall. EMOTE v1 is still available. Please see the demo script for details. + ## Installation diff --git a/inferno_apps/TalkingHead/demos/demo_eval_talking_head_on_audio.py b/inferno_apps/TalkingHead/demos/demo_eval_talking_head_on_audio.py index 0779ca83..3844da95 100644 --- a/inferno_apps/TalkingHead/demos/demo_eval_talking_head_on_audio.py +++ b/inferno_apps/TalkingHead/demos/demo_eval_talking_head_on_audio.py @@ -104,7 +104,7 @@ def main(): parser.add_argument('--path_to_audio', type=str, default= str(get_path_to_assets() / "data/EMOTE_test_example_data/01_gday.wav")) parser.add_argument('--output_folder', type=str, default="results", help="Output folder to save the results to.") # parser.add_argument('--model_name', type=str, default='EMOTE', help='Name of the model to use.') - parser.add_argument('--model_name', type=str, default='EMOTEv2', help='Name of the model to use.') + parser.add_argument('--model_name', type=str, default='EMOTE_v2', help='Name of the model to use.') parser.add_argument('--path_to_models', type=str, default=str(get_path_to_assets() / "TalkingHead/models")) parser.add_argument('--save_video', type=bool, default=True, help="If true, output images will be saved") parser.add_argument('--save_flame', type=bool, default=False, help="If true, output FLAME values for shape, expression, jaw pose will be saved")