Commit ·
b26ff1b
1
Parent(s): cec58ea
Added more logging messages
Browse files- handler.py +2 -1
handler.py
CHANGED
|
@@ -200,7 +200,7 @@ class EndpointHandler:
|
|
| 200 |
frame_embeddings, video_metadata = future.result()
|
| 201 |
if frame_embeddings is not None:
|
| 202 |
processed_video_embeddings.append(frame_embeddings)
|
| 203 |
-
|
| 204 |
processed_metadata = {
|
| 205 |
"text": video_metadata["caption"],
|
| 206 |
"source": video_metadata["url"],
|
|
@@ -208,6 +208,7 @@ class EndpointHandler:
|
|
| 208 |
**video_metadata,
|
| 209 |
}
|
| 210 |
processed_videos_metadata.append(processed_metadata)
|
|
|
|
| 211 |
self.logger.info(f"Finished processing batch {i}")
|
| 212 |
# Return the embeddings
|
| 213 |
self.logger.info("Returning embeddings and metadata of all batches")
|
|
|
|
| 200 |
frame_embeddings, video_metadata = future.result()
|
| 201 |
if frame_embeddings is not None:
|
| 202 |
processed_video_embeddings.append(frame_embeddings)
|
| 203 |
+
self.logger.info("Finished appending video embedding")
|
| 204 |
processed_metadata = {
|
| 205 |
"text": video_metadata["caption"],
|
| 206 |
"source": video_metadata["url"],
|
|
|
|
| 208 |
**video_metadata,
|
| 209 |
}
|
| 210 |
processed_videos_metadata.append(processed_metadata)
|
| 211 |
+
self.logger.info("Finished appending video metadata")
|
| 212 |
self.logger.info(f"Finished processing batch {i}")
|
| 213 |
# Return the embeddings
|
| 214 |
self.logger.info("Returning embeddings and metadata of all batches")
|