Spaces:
Sleeping
Sleeping
Upload app.py
Browse files
app.py
CHANGED
|
@@ -4,7 +4,7 @@ from tensorflow.keras.applications.inception_resnet_v2 import preprocess_input
|
|
| 4 |
from tensorflow.keras.preprocessing import image
|
| 5 |
import numpy as np
|
| 6 |
|
| 7 |
-
|
| 8 |
|
| 9 |
# Function for prediction
|
| 10 |
def predict(img):
|
|
|
|
| 4 |
from tensorflow.keras.preprocessing import image
|
| 5 |
import numpy as np
|
| 6 |
|
| 7 |
+
model = tf.keras.models.load_model("inceptionv3_model_with_custom_layer.h5")
|
| 8 |
|
| 9 |
# Function for prediction
|
| 10 |
def predict(img):
|