juanpablomesa
commited on
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")
|