ccclemenfff commited on
Commit
eecb9b2
·
1 Parent(s): a30f6bf
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -111,7 +111,7 @@ class EndpointHandler:
111
  ])
112
  return transform(image)
113
 
114
- def _load_video(self, video_bytes: bytes, num_segments=16) -> torch.Tensor:
115
  with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as tmpfile:
116
  tmpfile.write(video_bytes)
117
  video_path = tmpfile.name
 
111
  ])
112
  return transform(image)
113
 
114
+ def _load_video(self, video_bytes: bytes, num_segments=8) -> torch.Tensor:
115
  with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as tmpfile:
116
  tmpfile.write(video_bytes)
117
  video_path = tmpfile.name