Update app.py

app.py

@@ -7,7 +7,7 @@ import tensorflow as tf
 
 model = keras.models.load_model('my_model (2).h5')
 
-CLASS_NAMES = ['Necrotic Tumor', 'Non-Tumor', 'Viable Tumor']
+CLASS_NAMES = ['Non-Tumor', 'Non-Viable-Tumor', 'Viable', 'viable: non-viable']
 
 def make_gradcam_heatmap(img_array, model):
     last_conv_layer = model.get_layer('last_conv_layer')
@@ -68,7 +68,7 @@ demo = gr.Interface(
         gr.Textbox(label="Classification Result")
     ],
     title="Bone Cancer Detection (Osteosarcoma)",
-    description="Upload an H&E stained histopathology image to classify as Non-Tumor, Viable Tumor, or Necrotic Tumor."
+    description="Upload an H&E stained histopathology image to classify as Non-Tumor, Non-Viable-Tumor, Viable, or viable: non-viable."
 )
 
 demo.launch()