Update app.py
Browse files
app.py
CHANGED
@@ -32,7 +32,7 @@ def classify_image(image):
|
|
32 |
# Convert the index to the model's class label
|
33 |
label_pneumonia = index_to_label_pneumonia.get(predicted_class_idx_pneumonia, "Unknown Label")
|
34 |
label_tuberculosis = index_to_label_tuberculosis.get(predicted_class_idx_tuberculosis, "Unknown Label")
|
35 |
-
label = label_pneumonia+"
|
36 |
|
37 |
return label
|
38 |
|
|
|
32 |
# Convert the index to the model's class label
|
33 |
label_pneumonia = index_to_label_pneumonia.get(predicted_class_idx_pneumonia, "Unknown Label")
|
34 |
label_tuberculosis = index_to_label_tuberculosis.get(predicted_class_idx_tuberculosis, "Unknown Label")
|
35 |
+
label = label_pneumonia+" "+label_tuberculosis
|
36 |
|
37 |
return label
|
38 |
|