Update app_celery.py

app_celery.py

@@ -30,6 +30,8 @@ from celery import Celery
 # from gevent import monkey
 # monkey.patch_all()
 import torch.multiprocessing as t
+import multiprocessing
+multiprocessing.set_start_method('spawn', force=True)
 
 class AnimationConfig:
     def __init__(self, driven_audio_path, source_image_path, result_folder,pose_style,expression_scale,enhancer,still,preprocess,ref_pose_video_path, image_hardcoded):
@@ -63,7 +65,7 @@ class AnimationConfig:
         self.camera_d = 10.
         self.z_near = 5.
         self.z_far = 15.
-        self.device = 'cuda'
+        self.device = 'cpu'
         self.image_hardcoded = image_hardcoded
 
 
@@ -564,5 +566,4 @@ def health_status():
     response = {"online": "true"}
     return jsonify(response)
 if __name__ == '__main__':
-    t.set_start_method('spawn', force=True)
     app.run(debug=True)