diff --git a/frozen_in_time.py b/frozen_in_time.py
index ac69ef9..9978b7a 100644
--- a/frozen_in_time.py
+++ b/frozen_in_time.py
@@ -14,7 +14,7 @@
 import numpy
 import numpy as np
 import torch
-
+import os
 from typing import List, Union
 from torchvision import transforms
 from towhee.models import frozen_in_time