summarize_video main function

This commit is contained in:
Cemal Emre Albayrak 2024-04-27 18:27:06 +03:00
parent 41d8665300
commit e9a7d7c785

View File

@ -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()