mednow commited on
Commit
c18f20e
·
verified ·
1 Parent(s): 55cc86e

Upload 2 files

Browse files
Files changed (3) hide show
  1. .gitattributes +1 -0
  2. app.py +4 -4
  3. 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('best_model_.keras')
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 # Use decimal percentage
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": "best_model_.keras",
51
- "accuracy": 90.0
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