Update app.py

app.py

@@ -44,8 +44,8 @@ def classify_image_tuberculosis(image):
     predicted_class_idx = logits.argmax(-1).item()
     # Define a manual mapping of label indices to human-readable labels
     index_to_label = {
-        0: "NORMAL",
-        1: "PNEUMONIA"
+        0: "PNEUMONIA = NO",
+        1: "PNEUMONIA = YES"
     }
 
     # Convert the index to the model's class label
@@ -69,8 +69,8 @@ def make_block(dem):
                     in_prompt_2 = gr.Image()
                     out_response_2 = gr.Label()
                 b2 = gr.Button("Enter")
-            b1.click(classify_image_pneumonia, inputs=in_prompt_1, outputs=out_response_1)
-            b2.click(classify_image_tuberculosis, inputs=in_prompt_2, outputs=out_response_2)
+            b1.Block(classify_image_pneumonia, inputs=in_prompt_1, outputs=out_response_1)
+            b2.Block(classify_image_tuberculosis, inputs=in_prompt_2, outputs=out_response_2)
            
 if __name__ == '__main__':