Feature request: added json config
This commit is contained in:
@@ -1,20 +1,36 @@
|
||||
import os
|
||||
import json
|
||||
import argparse
|
||||
|
||||
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"]
|
||||
# Create a parser
|
||||
parser = argparse.ArgumentParser(description='Transcript Generator for Videos')
|
||||
|
||||
# Add the arguments
|
||||
parser.add_argument('--job', type=str, help='The name of the job', required=True)
|
||||
|
||||
# Parse the arguments
|
||||
args = parser.parse_args()
|
||||
|
||||
# Load the job config
|
||||
with open(f'jobs/{args.job}.json', 'r') as f:
|
||||
job_config = json.load(f)
|
||||
|
||||
# Check the hf_token
|
||||
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)
|
||||
transcript_generator = TranscriptGenerator(
|
||||
name=job_config['name'],
|
||||
language_code=job_config['language_code']
|
||||
)
|
||||
|
||||
transcript_generator.generate(job_config['source'])
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
|
||||
Reference in New Issue
Block a user