Spaces:
Sleeping
Sleeping
update_name_model_class
Browse files
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("
|
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'}
|