Update main.py

main.py

@@ -4,6 +4,7 @@ import argparse
 import io
 import json
 import os
+import platform
 import re
 from typing import Dict, List