Spanicin commited on
Commit
84dc8e6
·
verified ·
1 Parent(s): aa16990

updated app.py with eye and pose parameters

Browse files
Files changed (1) hide show
  1. app.py +5 -3
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 = None
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['ref_pose']
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
- with tempfile.NamedTemporaryFile(suffix=".mp4", prefix="ref_pose_",dir=TEMP_DIR.name, delete=False) as temp_file:
 
 
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)