ceperaltab's picture
Upload folder using huggingface_hub
6c0db32 verified
raw
history blame contribute delete
496 Bytes
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()