root
commited on
Commit
·
5066b17
1
Parent(s):
2cfffe0
python to python3
Browse files- handler.py +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"
|
| 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'
|
| 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
|