add

Signed-off-by: xujinling <jinling.xu@zilliz.com>
3 years ago · a6af44a913
1 changed files with 1 additions and 1 deletions
--- 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