Spaces:

sg7849
/

ml_rag_assistant

Sleeping

sg7849 commited on Apr 27

Commit

30811f7

verified ·

1 Parent(s): ebcd31f

accessing hf vids

Files changed (1) hide show

hf_retrieval.py CHANGED Viewed

@@ -90,7 +90,7 @@ def merge_chunks(chunks):
 def get_video_segment(video_id, start_time, end_time):
-    dataset = load_dataset("aegean-ai/ai-lectures-spring-24", split="train", streaming=True)
     for sample in dataset:
         if sample["__key__"] == video_id:
             break

 def get_video_segment(video_id, start_time, end_time):
+    dataset = load_dataset("aegean-ai/ai-lectures-spring-24", split="train", streaming=True, use_auth_token=os.environ["HF_API_TOKEN"])
     for sample in dataset:
         if sample["__key__"] == video_id:
             break