File size: 496 Bytes
6c0db32
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
import subprocess
import sys
import os

def install_dependencies():
    print("Installing dependencies...")
    # Use -v for verbose output so user sees progress
    subprocess.check_call([sys.executable, "-m", "pip", "install", "-v", "-r", "requirements_train.txt"])

def main():
    install_dependencies()
    print("Dependencies installed. Starting training...")
    # Import train only after dependencies are installed
    import train
    train.main()

if __name__ == "__main__":
    main()