juanpablomesa commited on
Commit
b26ff1b
·
1 Parent(s): cec58ea

Added more logging messages

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