cos518 COS 518 course project Test command (simple counting, not ML): python main.py --model_file sanity_test.py