mirror of
https://github.com/tcsenpai/easy-web-summarizer.git
synced 2025-06-06 18:45:21 +00:00
summarize_video main function
This commit is contained in:
parent
41d8665300
commit
e9a7d7c785
@ -28,13 +28,14 @@ def yt_summarization_chain():
|
|||||||
summarize_chain = load_summarize_chain(llm=llm, chain_type="refine", verbose=True)
|
summarize_chain = load_summarize_chain(llm=llm, chain_type="refine", verbose=True)
|
||||||
return summarize_chain
|
return summarize_chain
|
||||||
|
|
||||||
if __name__ == "__main__":
|
def summarize_video(video_link):
|
||||||
videos = ["https://www.youtube.com/watch?v=bYjQ9fzinT8", "https://www.youtube.com/watch?v=QCg0axyXxs4"]
|
transcript = get_transcript(video_link)
|
||||||
|
|
||||||
transcript = get_transcript(videos[0])
|
|
||||||
chunks = split_chunks(transcript)
|
chunks = split_chunks(transcript)
|
||||||
|
|
||||||
sum_chain = yt_summarization_chain()
|
sum_chain = yt_summarization_chain()
|
||||||
result = sum_chain.run(chunks)
|
result = sum_chain.run(chunks)
|
||||||
|
|
||||||
print(result)
|
return result
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
#summarize_video()
|
Loading…
x
Reference in New Issue
Block a user