root
commited on
Commit
·
54edee1
1
Parent(s):
5066b17
hint for python3
Browse files- handler.py +2 -1
handler.py
CHANGED
|
@@ -1,3 +1,5 @@
|
|
|
|
|
|
|
|
| 1 |
from typing import Dict, Any
|
| 2 |
import torch
|
| 3 |
from PIL import Image
|
|
@@ -318,7 +320,6 @@ class EndpointHandler():
|
|
| 318 |
pose_video_path = os.path.join(save_dir, "downloaded_video.mp4")
|
| 319 |
|
| 320 |
if not os.path.exists(pose_video_path):
|
| 321 |
-
print(f"Error running extract_dwpose_from_vid.py: {result.stderr}")
|
| 322 |
print("Contents of the temporary directory:")
|
| 323 |
self.print_directory_contents(temp_dir)
|
| 324 |
raise FileNotFoundError(f"The pose video was not found at: {pose_video_path}")
|
|
|
|
| 1 |
+
#!/usr/bin/env python3
|
| 2 |
+
|
| 3 |
from typing import Dict, Any
|
| 4 |
import torch
|
| 5 |
from PIL import Image
|
|
|
|
| 320 |
pose_video_path = os.path.join(save_dir, "downloaded_video.mp4")
|
| 321 |
|
| 322 |
if not os.path.exists(pose_video_path):
|
|
|
|
| 323 |
print("Contents of the temporary directory:")
|
| 324 |
self.print_directory_contents(temp_dir)
|
| 325 |
raise FileNotFoundError(f"The pose video was not found at: {pose_video_path}")
|