jarvis-cloud / modules /device_detect.py
Jarvis2345's picture
Upload folder using huggingface_hub
2dae9bb verified
Raw
History Blame Contribute Delete
771 Bytes
import platform
import psutil
import shutil
def detect_device():
battery = psutil.sensors_battery()
ram = round(psutil.virtual_memory().total / (1024**3))
disk = shutil.disk_usage("C:\\")
free = round(disk.free / (1024**3))
data = {
"system": platform.system(),
"release": platform.release(),
"machine": platform.machine(),
"ram": ram,
"free_disk": free,
"battery": battery is not None
}
if battery:
data["type"] = "laptop"
else:
data["type"] = "desktop"
if ram >= 16:
data["class"] = "high performance"
elif ram >= 8:
data["class"] = "standard"
else:
data["class"] = "lightweight"
return data