Spaces:
Sleeping
Sleeping
Jon Solow
commited on
Commit
·
5f9a45d
1
Parent(s):
5c3d0ab
Update tf import paths
Browse files- src/handler.py +1 -1
- 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
|
3 |
from config import MODEL_HDF5_PATH
|
4 |
|
5 |
|
6 |
-
|
|
|
|
|
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
|
14 |
-
from keras.layers
|
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()
|