updated app.py with eye and pose parameters
Browse files
app.py
CHANGED
|
@@ -28,7 +28,7 @@ class AnimationConfig:
|
|
| 28 |
def __init__(self, driven_audio_path, source_image_path, result_folder,pose_style,expression_scale,enhancer,still,preprocess,ref_pose_video_path):
|
| 29 |
self.driven_audio = driven_audio_path
|
| 30 |
self.source_image = source_image_path
|
| 31 |
-
self.ref_eyeblink =
|
| 32 |
self.ref_pose = ref_pose_video_path
|
| 33 |
self.checkpoint_dir = './checkpoints'
|
| 34 |
self.result_dir = result_folder
|
|
@@ -213,7 +213,7 @@ def generate_video():
|
|
| 213 |
print('still', still)
|
| 214 |
preprocess = request.form.get('preprocess', 'crop')
|
| 215 |
print('preprocess selected: ',preprocess)
|
| 216 |
-
ref_pose_video = request.files
|
| 217 |
|
| 218 |
if target_language != 'original_text':
|
| 219 |
response = translate_text(text_prompt, target_language)
|
|
@@ -265,7 +265,9 @@ def generate_video():
|
|
| 265 |
result_folder = os.path.join(save_dir, "results")
|
| 266 |
os.makedirs(result_folder, exist_ok=True)
|
| 267 |
|
| 268 |
-
|
|
|
|
|
|
|
| 269 |
ref_pose_video_path = temp_file.name
|
| 270 |
ref_pose_video.save(ref_pose_video_path)
|
| 271 |
print('ref_pose_video_path',ref_pose_video_path)
|
|
|
|
| 28 |
def __init__(self, driven_audio_path, source_image_path, result_folder,pose_style,expression_scale,enhancer,still,preprocess,ref_pose_video_path):
|
| 29 |
self.driven_audio = driven_audio_path
|
| 30 |
self.source_image = source_image_path
|
| 31 |
+
self.ref_eyeblink = ref_pose_video_path
|
| 32 |
self.ref_pose = ref_pose_video_path
|
| 33 |
self.checkpoint_dir = './checkpoints'
|
| 34 |
self.result_dir = result_folder
|
|
|
|
| 213 |
print('still', still)
|
| 214 |
preprocess = request.form.get('preprocess', 'crop')
|
| 215 |
print('preprocess selected: ',preprocess)
|
| 216 |
+
ref_pose_video = request.files.get('ref_pose', None)
|
| 217 |
|
| 218 |
if target_language != 'original_text':
|
| 219 |
response = translate_text(text_prompt, target_language)
|
|
|
|
| 265 |
result_folder = os.path.join(save_dir, "results")
|
| 266 |
os.makedirs(result_folder, exist_ok=True)
|
| 267 |
|
| 268 |
+
ref_pose_video_path = None
|
| 269 |
+
if ref_pose_video:
|
| 270 |
+
with tempfile.NamedTemporaryFile(suffix=".mp4", prefix="ref_pose_",dir=TEMP_DIR.name, delete=False) as temp_file:
|
| 271 |
ref_pose_video_path = temp_file.name
|
| 272 |
ref_pose_video.save(ref_pose_video_path)
|
| 273 |
print('ref_pose_video_path',ref_pose_video_path)
|