cache
Browse files
app.py
CHANGED
|
@@ -24,10 +24,19 @@ st.markdown(
|
|
| 24 |
unsafe_allow_html=True
|
| 25 |
)
|
| 26 |
|
| 27 |
-
|
| 28 |
-
|
| 29 |
-
|
| 30 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 31 |
|
| 32 |
# CNN model
|
| 33 |
|
|
|
|
| 24 |
unsafe_allow_html=True
|
| 25 |
)
|
| 26 |
|
| 27 |
+
@st.cache_resource
|
| 28 |
+
def load_models():
|
| 29 |
+
# Load all models at once
|
| 30 |
+
eff_net_model = tf.keras.models.load_model('EfficientNet_Models/efficientnetb3_binary_classifier_8.h5')
|
| 31 |
+
eff_net_art_model = tf.keras.models.load_model('EfficientNet_Models/EfficientNet_fine_tune_art_model.h5')
|
| 32 |
+
cnn_model = tf.keras.models.load_model('CNN_model_weight/model_weights.weights.h5')
|
| 33 |
+
return eff_net_model, eff_net_art_model, cnn_model
|
| 34 |
+
|
| 35 |
+
eff_net_model, eff_net_art_model, cnn_model = load_models()
|
| 36 |
+
|
| 37 |
+
# eff_net_model = tf.keras.models.load_model('EfficientNet_Models/efficientnetb3_binary_classifier_8.h5')
|
| 38 |
+
# eff_net_art_model = tf.keras.models.load_model('EfficientNet_Models/EfficientNet_fine_tune_art_model.h5')
|
| 39 |
+
# cnn_model = 'CNN_model_weight/model_weights.weights.h5'
|
| 40 |
|
| 41 |
# CNN model
|
| 42 |
|