diff --git a/byte.py b/byte.py
index d0a5a92f..23c2166c 100644
--- a/byte.py
+++ b/byte.py
@@ -5,6 +5,7 @@ from PIL import Image
 import numpy as np
 import torch
 import struct
+import magic
 
 
 from enum import auto