Create main.py

main.py

@@ -0,0 +1,3 @@
+import os
+os.system("chmod +x train.sh")
+os.system("./train.sh")