Update app_concurrent.py

app_concurrent.py

@@ -1,5 +1,4 @@
 from flask import Flask, request, jsonify, Response, stream_with_context
-import torch
 import shutil
 import os
 import sys
@@ -391,6 +390,7 @@ def split_audio(audio_path, chunk_duration=5):
 
 
 def process_video_for_chunk_sync(audio_chunk_path, args_dict, chunk_index):
+    import torch
     """
     Synchronous function to process a video chunk. This will be submitted to concurrent.futures ProcessPoolExecutor.
     """
@@ -517,8 +517,7 @@ def generate_video():
     'still': still,
     'preprocess': preprocess,
     'ref_pose_video_path': ref_pose_video_path,
-    'image_hardcoded': image_hardcoded,
-    'device': 'cuda' if torch.cuda.is_available() else 'cpu'}
+    'image_hardcoded': image_hardcoded}
 
     executor = concurrent.futures.ProcessPoolExecutor(max_workers=MAX_WORKERS)    
     try: