bug fixed
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ import tensorflow as tf
|
|
4 |
|
5 |
#model = joblib.load('model.pkl')
|
6 |
model = tf.keras.models.load_model('best_model.h5')
|
7 |
-
categories = ["Normal",
|
8 |
|
9 |
def classify(img):
|
10 |
img = img.reshape((-1, 224, 224, 3))
|
|
|
4 |
|
5 |
#model = joblib.load('model.pkl')
|
6 |
model = tf.keras.models.load_model('best_model.h5')
|
7 |
+
categories = ["Normal","Pneumonia", "Tubercolosis"]
|
8 |
|
9 |
def classify(img):
|
10 |
img = img.reshape((-1, 224, 224, 3))
|