Update handler.py
Browse files- handler.py +6 -2
handler.py
CHANGED
@@ -18,9 +18,13 @@ class EndpointHandler:
|
|
18 |
self.model.eval()
|
19 |
|
20 |
def __call__(self, data):
|
21 |
-
video_path = data["inputs"]["video_path"]
|
22 |
# ftp_password = data["inputs"].get("ftp_password")
|
23 |
-
ftp_password = data["inputs"]["ftp_password"]
|
|
|
|
|
|
|
|
|
24 |
|
25 |
processed_frames = read_video(video_path, ftp_password)
|
26 |
|
|
|
18 |
self.model.eval()
|
19 |
|
20 |
def __call__(self, data):
|
21 |
+
# video_path = data["inputs"]["video_path"]
|
22 |
# ftp_password = data["inputs"].get("ftp_password")
|
23 |
+
# ftp_password = data["inputs"]["ftp_password"]
|
24 |
+
|
25 |
+
inputs = json.loads(data)
|
26 |
+
video_path = inputs.get("inputs", {}).get("video_path", "")
|
27 |
+
ftp_password = inputs.get("inputs", {}).get("ftp_password", "")
|
28 |
|
29 |
processed_frames = read_video(video_path, ftp_password)
|
30 |
|