Spaces:
Sleeping
Sleeping
Upload 2 files
Browse files- .gitattributes +1 -0
- app.py +4 -4
- best_model.keras +3 -0
.gitattributes
CHANGED
|
@@ -34,3 +34,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
best_model_.keras filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
best_model_.keras filter=lfs diff=lfs merge=lfs -text
|
| 37 |
+
best_model.keras filter=lfs diff=lfs merge=lfs -text
|
app.py
CHANGED
|
@@ -24,7 +24,7 @@ def preprocess_image(img_path, img_height, img_width, model_type="CNN"):
|
|
| 24 |
# Load the Keras model
|
| 25 |
@st.cache_resource
|
| 26 |
def load_keras_model():
|
| 27 |
-
return tf.keras.models.load_model('
|
| 28 |
|
| 29 |
# Load the other models
|
| 30 |
@st.cache_resource
|
|
@@ -36,7 +36,7 @@ def load_pickle_model(model_path):
|
|
| 36 |
models_info = {
|
| 37 |
"ANN Model": {
|
| 38 |
"path": "ann_model.pkl",
|
| 39 |
-
"accuracy": 71.0
|
| 40 |
},
|
| 41 |
"Decision Tree": {
|
| 42 |
"path": "decision_tree_classifier_model.pkl",
|
|
@@ -47,8 +47,8 @@ models_info = {
|
|
| 47 |
"accuracy": 71.47
|
| 48 |
},
|
| 49 |
"CNN Model": {
|
| 50 |
-
"path": "
|
| 51 |
-
"accuracy":
|
| 52 |
}
|
| 53 |
}
|
| 54 |
|
|
|
|
| 24 |
# Load the Keras model
|
| 25 |
@st.cache_resource
|
| 26 |
def load_keras_model():
|
| 27 |
+
return tf.keras.models.load_model('best_model.keras')
|
| 28 |
|
| 29 |
# Load the other models
|
| 30 |
@st.cache_resource
|
|
|
|
| 36 |
models_info = {
|
| 37 |
"ANN Model": {
|
| 38 |
"path": "ann_model.pkl",
|
| 39 |
+
"accuracy": 71.0
|
| 40 |
},
|
| 41 |
"Decision Tree": {
|
| 42 |
"path": "decision_tree_classifier_model.pkl",
|
|
|
|
| 47 |
"accuracy": 71.47
|
| 48 |
},
|
| 49 |
"CNN Model": {
|
| 50 |
+
"path": "best_model.keras",
|
| 51 |
+
"accuracy": 84.61
|
| 52 |
}
|
| 53 |
}
|
| 54 |
|
best_model.keras
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:30b64a31f5f3c3f17f837ade4f4965ec4a615e302ea54ca94fbe9386a4dec4ef
|
| 3 |
+
size 1147997225
|