juanpablomesa commited on
Commit
f33eeea
·
1 Parent(s): 0dbfcb8

Changed set for list to be json serializable

Browse files
Files changed (1) hide show
  1. handler.py +3 -3
handler.py CHANGED
@@ -190,10 +190,10 @@ class EndpointHandler:
190
  self.logger.info("Extracting text from frames.")
191
  text_extraction_start_time = timeit.default_timer()
192
  frame_texts = [self.reader.readtext(frame, detail=0) for frame in frames]
193
- texts_set = set()
194
  for text_list in frame_texts:
195
- [texts_set.add(text) for text in text_list]
196
- video_metadata["extracted_text"] = texts_set
197
  text_extraction_end_time = timeit.default_timer()
198
  self.logger.info(
199
  f"Text extraction took {text_extraction_end_time - text_extraction_start_time} seconds"
 
190
  self.logger.info("Extracting text from frames.")
191
  text_extraction_start_time = timeit.default_timer()
192
  frame_texts = [self.reader.readtext(frame, detail=0) for frame in frames]
193
+ all_texts_list = []
194
  for text_list in frame_texts:
195
+ [all_texts_list.append(text) for text in text_list]
196
+ video_metadata["extracted_text"] = all_texts_list
197
  text_extraction_end_time = timeit.default_timer()
198
  self.logger.info(
199
  f"Text extraction took {text_extraction_end_time - text_extraction_start_time} seconds"