Spanicin commited on
Commit
0aa58cc
·
verified ·
1 Parent(s): 6d6a4a5

Update app_celery.py

Browse files
Files changed (1) hide show
  1. app_celery.py +2 -2
app_celery.py CHANGED
@@ -29,7 +29,7 @@ import pickle
29
  from celery import Celery
30
  # from gevent import monkey
31
  # monkey.patch_all()
32
- import torch.multiprocessing as mp
33
 
34
  class AnimationConfig:
35
  def __init__(self, driven_audio_path, source_image_path, result_folder,pose_style,expression_scale,enhancer,still,preprocess,ref_pose_video_path, image_hardcoded):
@@ -564,5 +564,5 @@ def health_status():
564
  response = {"online": "true"}
565
  return jsonify(response)
566
  if __name__ == '__main__':
567
- mp.set_start_method('spawn', force=True)
568
  app.run(debug=True)
 
29
  from celery import Celery
30
  # from gevent import monkey
31
  # monkey.patch_all()
32
+ import torch.multiprocessing as t
33
 
34
  class AnimationConfig:
35
  def __init__(self, driven_audio_path, source_image_path, result_folder,pose_style,expression_scale,enhancer,still,preprocess,ref_pose_video_path, image_hardcoded):
 
564
  response = {"online": "true"}
565
  return jsonify(response)
566
  if __name__ == '__main__':
567
+ t.set_start_method('spawn', force=True)
568
  app.run(debug=True)