vikramvasudevan commited on
Commit
96fa70b
·
verified ·
1 Parent(s): 05ac23c

Upload folder using huggingface_hub

Browse files
modules/youtube_metadata/youtube_poller.py CHANGED
@@ -70,7 +70,7 @@ def add_to_chroma(collection: Collection, new_videos):
70
  return
71
  count = collection.count()
72
  logger.info("new_videos = %s", new_videos)
73
- documents = [f"{v['video_title']} - v['description']" for v in new_videos]
74
  embeddings = [get_embedding(doc) for doc in documents]
75
  collection.add(
76
  documents=documents,
 
70
  return
71
  count = collection.count()
72
  logger.info("new_videos = %s", new_videos)
73
+ documents = [f"{v['video_title']} - {v['description']}" for v in new_videos]
74
  embeddings = [get_embedding(doc) for doc in documents]
75
  collection.add(
76
  documents=documents,