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
|