21 lines
500 B
Python
21 lines
500 B
Python
import os
|
|
|
|
from src.transcript_generator import TranscriptGenerator
|
|
|
|
|
|
def main():
|
|
# input could be a YouTube link or a local file
|
|
input_sources = ["https://www.youtube.com/watch?v=BVVo1BaL6JA"]
|
|
|
|
hf_token = os.environ.get("HF_TOKEN")
|
|
if not hf_token:
|
|
hf_token = input("Please enter your HF_TOKEN: ")
|
|
os.environ["HF_TOKEN"] = hf_token
|
|
|
|
transcript_generator = TranscriptGenerator()
|
|
transcript_generator.generate(input_sources)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|