Commit
·
eecb9b2
1
Parent(s):
a30f6bf
16->8
Browse files- 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=
|
| 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
|