iloncka commited on
Commit
07e3301
1 Parent(s): b59e813

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -15
utils.py CHANGED
@@ -6,26 +6,13 @@ import os.path as op
6
  import pickle
7
 
8
 
9
- from contextlib import contextmanager
10
- import pathlib
11
-
12
- @contextmanager
13
- def set_posix_windows():
14
- posix_backup = pathlib.PosixPath
15
- try:
16
- pathlib.PosixPath = pathlib.WindowsPath
17
- yield
18
- finally:
19
- pathlib.PosixPath = posix_backup
20
-
21
  HOME_DIR = os.getcwd()
22
  DATA_DIR = op.join(HOME_DIR, 'data')
23
  MODELS_DIR = op.join(HOME_DIR,'models')
24
 
25
 
26
- def load_model():
27
- with set_posix_windows():
28
- learn = load_learner(op.join(MODELS_DIR, "vit_tiny_patch16_224_2.pkl"))
29
  return learn
30
 
31
  # learn = load_learner(op.join(MODELS_DIR, "vit_tiny_patch16_224_2.pkl"))
 
6
  import pickle
7
 
8
 
 
 
 
 
 
 
 
 
 
 
 
 
9
  HOME_DIR = os.getcwd()
10
  DATA_DIR = op.join(HOME_DIR, 'data')
11
  MODELS_DIR = op.join(HOME_DIR,'models')
12
 
13
 
14
+ def load_model():
15
+ learn = load_learner(op.join(MODELS_DIR, "vit_tiny_patch16_224_2.pkl"))
 
16
  return learn
17
 
18
  # learn = load_learner(op.join(MODELS_DIR, "vit_tiny_patch16_224_2.pkl"))