Jon Solow commited on
Commit
5f9a45d
·
1 Parent(s): 5c3d0ab

Update tf import paths

Browse files
Files changed (2) hide show
  1. src/handler.py +1 -1
  2. src/model.py +8 -6
src/handler.py CHANGED
@@ -3,7 +3,7 @@ from skimage.io import imread
3
  from io import BytesIO
4
  import numpy as np
5
  import urllib
6
- from keras.preprocessing.image import array_to_img, img_to_array
7
 
8
  from PIL import Image
9
 
 
3
  from io import BytesIO
4
  import numpy as np
5
  import urllib
6
+ from tensorflow.keras.preprocessing.image import array_to_img, img_to_array
7
 
8
  from PIL import Image
9
 
src/model.py CHANGED
@@ -1,17 +1,19 @@
1
  import numpy as np
2
- from keras.utils.data_utils import get_file
3
  from config import MODEL_HDF5_PATH
4
 
5
 
6
- LABELS = np.loadtxt("labels.txt", dtype=object, delimiter="\n")
 
 
7
 
8
 
9
  def initialize_model():
10
  # import the necessary packages
11
- from keras.models import Sequential
12
- from keras.layers import BatchNormalization
13
- from keras.layers.convolutional import Conv2D, MaxPooling2D
14
- from keras.layers.core import Flatten, Dropout, Dense
15
 
16
  # CONV => RELU => POOL
17
  cnn = Sequential()
 
1
  import numpy as np
2
+ from tensorflow.keras.utils import get_file
3
  from config import MODEL_HDF5_PATH
4
 
5
 
6
+ with open("labels.txt", "r") as f:
7
+ LABELS = list(filter(None, f.read().split("\n")))
8
+
9
 
10
 
11
  def initialize_model():
12
  # import the necessary packages
13
+ from tensorflow.keras.models import Sequential
14
+ from tensorflow.keras.layers import BatchNormalization
15
+ from tensorflow.keras.layers import Conv2D, MaxPooling2D
16
+ from tensorflow.keras.layers import Flatten, Dropout, Dense
17
 
18
  # CONV => RELU => POOL
19
  cnn = Sequential()