Nina-HK commited on
Commit
02b49f6
1 Parent(s): 55fb13e

update_name_model_class

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ from transformers import pipeline
28
  model_binary = tf.keras.models.load_model("CNN_binary")
29
 
30
  # load the multi-label classification model
31
- model_multi = tf.keras.models.load_model("CNN_multi")
32
 
33
  # define the labels for the multi-label classification model
34
  labels_multi = {0: 'healthy', 1: 'mild', 2: 'moderate'}
 
28
  model_binary = tf.keras.models.load_model("CNN_binary")
29
 
30
  # load the multi-label classification model
31
+ model_multi = tf.keras.models.load_model("CNN_multiclass")
32
 
33
  # define the labels for the multi-label classification model
34
  labels_multi = {0: 'healthy', 1: 'mild', 2: 'moderate'}