Added in new model names

This commit is contained in:
Lucas Oskorep
2019-04-26 16:18:00 -05:00
parent 24e8d71306
commit 601f4f7633
+1 -1
View File
@@ -1,7 +1,7 @@
from tensorflow.contrib.keras.api import keras
from tensorflow.contrib import lite
keras_file = "InceptionV3Full.hdf5"
keras_file = "mobilenet.hdf5"
keras.models.load_model(keras_file)
h5_model = keras.models.load_model(keras_file)