Commit
·
6747b77
1
Parent(s):
66eb7c3
Update prediction.py
Browse files- prediction.py +3 -3
prediction.py
CHANGED
|
@@ -6,9 +6,9 @@ import streamlit as st
|
|
| 6 |
from tensorflow.keras.models import load_model
|
| 7 |
import io
|
| 8 |
|
| 9 |
-
model_gender = load_model('model_gender.keras')
|
| 10 |
-
model_ethnicity = load_model('model_ethnicity.keras')
|
| 11 |
-
model_age = load_model('model_age.keras')
|
| 12 |
|
| 13 |
# Mengubah ukuran gambar dan mengonversinya ke mode grayscale
|
| 14 |
def preprocess_image(img,):
|
|
|
|
| 6 |
from tensorflow.keras.models import load_model
|
| 7 |
import io
|
| 8 |
|
| 9 |
+
model_gender = load_model('model_gender.keras', safe_mode=False)
|
| 10 |
+
model_ethnicity = load_model('model_ethnicity.keras', safe_mode=False)
|
| 11 |
+
model_age = load_model('model_age.keras', safe_mode=False)
|
| 12 |
|
| 13 |
# Mengubah ukuran gambar dan mengonversinya ke mode grayscale
|
| 14 |
def preprocess_image(img,):
|