37 lines
922 B
Python
37 lines
922 B
Python
import os
|
|
import json
|
|
import argparse
|
|
|
|
from src.transcript_generator import TranscriptGenerator
|
|
|
|
|
|
def main():
|
|
# 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(
|
|
name=job_config['name'],
|
|
language_code=job_config['language_code']
|
|
)
|
|
|
|
transcript_generator.generate(job_config['source'])
|
|
|
|
if __name__ == "__main__":
|
|
main()
|