Update app.py

app.py

@@ -27,8 +27,8 @@ def classify_image(image):
     predicted_class_idx_pneumonia = logits_pneumonia.argmax(-1).item()
     predicted_class_idx_tuberculosis = logits_tuberculosis.argmax(-1).item()
     # Define a manual mapping of label indices to human-readable labels
-    index_to_label_pneumonia = {0: "NORMAL",1: "PNEUMONIA"}
-    index_to_label_tuberculosis = {0: "NORMAL",1: "TUBERCULOSIS"}
+    index_to_label_pneumonia = {0: "Pneumonia = NO",1: "Pneumonia = YES"}
+    index_to_label_tuberculosis = {0: "Tuberculosis = NO",1: "Tuberculosis = YES"}
     # Convert the index to the model's class label
     label_pneumonia = index_to_label_pneumonia.get(predicted_class_idx_pneumonia, "Unknown Label")
     label_tuberculosis = index_to_label_tuberculosis.get(predicted_class_idx_tuberculosis, "Unknown Label")