Update app.py
Browse files
app.py
CHANGED
|
@@ -9,7 +9,7 @@ def predict(image_file):
|
|
| 9 |
img = np.expand_dims(img, axis=0)
|
| 10 |
img = img / 255
|
| 11 |
|
| 12 |
-
model = load_model("
|
| 13 |
result = model.predict(img)
|
| 14 |
|
| 15 |
if result[0][0] >= 0.5:
|
|
|
|
| 9 |
img = np.expand_dims(img, axis=0)
|
| 10 |
img = img / 255
|
| 11 |
|
| 12 |
+
model = load_model("vgg16_model.h5")
|
| 13 |
result = model.predict(img)
|
| 14 |
|
| 15 |
if result[0][0] >= 0.5:
|