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