root commited on
Commit
5066b17
·
1 Parent(s): 2cfffe0

python to python3

Browse files
Files changed (2) hide show
  1. handler.py +2 -2
  2. requirements.txt +0 -11
handler.py CHANGED
@@ -206,7 +206,7 @@ class EndpointHandler():
206
  base_dir = os.path.dirname(os.path.abspath(__file__))
207
  directory = os.path.join(base_dir, "Practical-RIFE", "inference_video.py")
208
  model_directory = os.path.join(base_dir, "Practical-RIFE", "train_log")
209
- command = f"python {directory} --video={video_path} --output={output_path} --multi={multi} --scale={scale} --model={model_directory}"
210
 
211
  # Run the command with shell=True
212
  result = subprocess.run(command, shell=True, capture_output=True, text=True)
@@ -305,7 +305,7 @@ class EndpointHandler():
305
  pose_output_path = os.path.join(temp_dir, "pose_videos")
306
 
307
  # Run the extract_dwpose_from_vid.py script
308
- command = f'python extract_dwpose_from_vid.py --video_root {video_root}'
309
 
310
  # Run the command with shell=True
311
  result = subprocess.run(command, shell=True, capture_output=True, text=True)
 
206
  base_dir = os.path.dirname(os.path.abspath(__file__))
207
  directory = os.path.join(base_dir, "Practical-RIFE", "inference_video.py")
208
  model_directory = os.path.join(base_dir, "Practical-RIFE", "train_log")
209
+ command = f"python3 {directory} --video={video_path} --output={output_path} --multi={multi} --scale={scale} --model={model_directory}"
210
 
211
  # Run the command with shell=True
212
  result = subprocess.run(command, shell=True, capture_output=True, text=True)
 
305
  pose_output_path = os.path.join(temp_dir, "pose_videos")
306
 
307
  # Run the extract_dwpose_from_vid.py script
308
+ command = f'python3 extract_dwpose_from_vid.py --video_root {video_root}'
309
 
310
  # Run the command with shell=True
311
  result = subprocess.run(command, shell=True, capture_output=True, text=True)
requirements.txt CHANGED
@@ -6,7 +6,6 @@ onnx==1.14.0
6
  insightface==0.7.3
7
  psutil==5.9.5
8
  tk==0.1.0
9
- pillow==9.5.0
10
  torch==2.0.1+cu118; sys_platform != 'darwin'
11
  torch==2.0.1; sys_platform == 'darwin'
12
  torchvision==0.15.2+cu118; sys_platform != 'darwin'
@@ -23,15 +22,12 @@ gfpgan==1.3.8
23
  gradio==3.41.2
24
  onnxruntime-coreml==1.13.1; python_version == '3.9' and sys_platform == 'darwin' and platform_machine != 'arm64'
25
  transformers==4.41.1
26
- controlnet-aux==0.0.7
27
 
28
  # Add additional dependencies
29
  diffusers==0.24.0
30
  omegaconf==2.2.3
31
 
32
  # Face swap related dependencies
33
- facenet-pytorch==2.5.2
34
- dlib==19.22.0
35
 
36
  # Stuff huggingface is complaining about
37
  einops==0.4.1
@@ -41,15 +37,8 @@ av==11.0.0
41
  accelerate==0.21.0
42
  clip @ https://github.com/openai/CLIP/archive/d50d76daa670286dd6cacf3bcd80b5e4823fc8e1.zip#sha256=b5842c25da441d6c581b53a5c60e0c2127ebafe0f746f8e15561a006c6c3be6a
43
  decord==0.6.0
44
- gradio_client==0.5.0
45
- imageio==2.33.0
46
- imageio-ffmpeg==0.4.9
47
- scikit-image==0.21.0
48
- scikit-learn==1.3.2
49
  scipy==1.11.4
50
  torchdiffeq==0.2.3
51
- torchmetrics==1.2.1
52
- torchsde==0.2.5
53
 
54
 
55
  # Additional dependencies for RIFE
 
6
  insightface==0.7.3
7
  psutil==5.9.5
8
  tk==0.1.0
 
9
  torch==2.0.1+cu118; sys_platform != 'darwin'
10
  torch==2.0.1; sys_platform == 'darwin'
11
  torchvision==0.15.2+cu118; sys_platform != 'darwin'
 
22
  gradio==3.41.2
23
  onnxruntime-coreml==1.13.1; python_version == '3.9' and sys_platform == 'darwin' and platform_machine != 'arm64'
24
  transformers==4.41.1
 
25
 
26
  # Add additional dependencies
27
  diffusers==0.24.0
28
  omegaconf==2.2.3
29
 
30
  # Face swap related dependencies
 
 
31
 
32
  # Stuff huggingface is complaining about
33
  einops==0.4.1
 
37
  accelerate==0.21.0
38
  clip @ https://github.com/openai/CLIP/archive/d50d76daa670286dd6cacf3bcd80b5e4823fc8e1.zip#sha256=b5842c25da441d6c581b53a5c60e0c2127ebafe0f746f8e15561a006c6c3be6a
39
  decord==0.6.0
 
 
 
 
 
40
  scipy==1.11.4
41
  torchdiffeq==0.2.3
 
 
42
 
43
 
44
  # Additional dependencies for RIFE