lanbogao commited on
Commit
b78af59
1 Parent(s): 68d802e

Update fetchYoutubeSubtitle.py

Browse files
Files changed (1) hide show
  1. fetchYoutubeSubtitle.py +1 -10
fetchYoutubeSubtitle.py CHANGED
@@ -5,8 +5,6 @@ import logging
5
 
6
  logger = logging.getLogger(__name__)
7
 
8
- from langchain.document_loaders import SRTLoader
9
- from langchain.text_splitter import RecursiveCharacterTextSplitter, TokenTextSplitter
10
  from typing import List, Dict
11
 
12
  SUBTITLE_DOWNLOADER_URL = 'https://savesubs.com'
@@ -96,11 +94,4 @@ async def fetchYoutubeSubtitle(videoId: str) -> Dict:
96
  }
97
  response = session.get(subtitleUrl)
98
 
99
- loader = SRTLoader(await response.blob())
100
- splitter = TokenTextSplitter(
101
- encodingName="cl100k_base",
102
- chunkSize=2048,
103
- )
104
-
105
- docs = await loader.loadAndSplit(splitter)
106
- return {"title": title, "docs": docs}
 
5
 
6
  logger = logging.getLogger(__name__)
7
 
 
 
8
  from typing import List, Dict
9
 
10
  SUBTITLE_DOWNLOADER_URL = 'https://savesubs.com'
 
94
  }
95
  response = session.get(subtitleUrl)
96
 
97
+ return {"title": title, "subtitle": response.text()}