Spaces:
Build error
Build error
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()
|