Spaces:

SocialAI
/

MIA-LungCancerDetect

Sleeping

SocialAI commited on May 30

Commit

932eb5f

verified ·

1 Parent(s): a2f2d80

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ import os
 model = tf.keras.models.load_model("fine_tuned_resnet50.h5")
 img_dim = (224, 224)
-lung_cancer_labels = ["Benign", "Malignant", "Normal"]
 # returning classifiers output
 def predict(img):

 model = tf.keras.models.load_model("fine_tuned_resnet50.h5")
 img_dim = (224, 224)
+lung_cancer_labels = ["Adenocarcinoma", "Benign", "Carcinoma"]
 # returning classifiers output
 def predict(img):