donghuna commited on
Commit
1a446c2
·
verified ·
1 Parent(s): 1f8df08

Update handler.py

Browse files
Files changed (1) hide show
  1. 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