Spaces:

Digital-Detectives
/

AI-vs-Real-Image-Detection

Sleeping

ff98 commited on Nov 29, 2024

Commit

ef8712b

1 Parent(s): 6ea15c9

model updated

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,19 +24,11 @@ st.markdown(
     unsafe_allow_html=True
 )
-@st.cache_resource
-def load_models():
-    # Load all models at once
-    eff_net_model = tf.keras.models.load_model('EfficientNet_Models/efficientnetb3_binary_classifier_8.h5')
-    eff_net_art_model = tf.keras.models.load_model('EfficientNet_Models/EfficientNet_fine_tune_art_model.h5')
-    cnn_model = tf.keras.models.load_model('CNN_model_weight/model_weights.weights.h5')
-    return eff_net_model, eff_net_art_model, cnn_model
-eff_net_model, eff_net_art_model, cnn_model = load_models()
-# eff_net_model = tf.keras.models.load_model('EfficientNet_Models/efficientnetb3_binary_classifier_8.h5')
-# eff_net_art_model = tf.keras.models.load_model('EfficientNet_Models/EfficientNet_fine_tune_art_model.h5')
-# cnn_model = 'CNN_model_weight/model_weights.weights.h5'
 # CNN model

     unsafe_allow_html=True
 )
+eff_net_model = tf.keras.models.load_model('EfficientNet_Models/efficientnetb3_binary_classifier_8.h5')
+eff_net_art_model = tf.keras.models.load_model('EfficientNet_Models/EfficientNet_fine_tune_art_model.h5')
+cnn_model = 'CNN_model_weight/model_weights.weights.h5'
 # CNN model