Spanicin commited on
Commit
076631b
·
verified ·
1 Parent(s): 8d04433

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -41,7 +41,7 @@ class AnimationConfig:
41
  self.checkpoint_dir = './checkpoints'
42
  self.result_dir = result_folder
43
  self.pose_style = pose_style
44
- self.batch_size = 32
45
  self.expression_scale = expression_scale
46
  self.input_yaw = None
47
  self.input_pitch = None
@@ -362,6 +362,7 @@ def generate_video():
362
  return jsonify({
363
  "message": "Video processed and saved successfully.",
364
  "video_url": video_url,
 
365
  "status": "success"
366
  })
367
  else:
 
41
  self.checkpoint_dir = './checkpoints'
42
  self.result_dir = result_folder
43
  self.pose_style = pose_style
44
+ self.batch_size = 128
45
  self.expression_scale = expression_scale
46
  self.input_yaw = None
47
  self.input_pitch = None
 
362
  return jsonify({
363
  "message": "Video processed and saved successfully.",
364
  "video_url": video_url,
365
+ "time_taken": time_taken
366
  "status": "success"
367
  })
368
  else: