Update app.py

app.py

@@ -7,7 +7,7 @@ import numpy as np
 model_path = "pneumonia_model_tl.keras"
 model = tf.keras.models.load_model(model_path)
  
-labels = ['Normal 1', 'Normal 2', 'Pneumonia 1', 'Pneumonia 2']
+labels = ['Normal', 'Pneumonia']
  
 # Define regression function
 def predict_regression(image):
@@ -27,6 +27,6 @@ output_text = gr.Textbox(label="Predicted Value")
 interface = gr.Interface(fn=predict_regression, 
                          inputs=input_image, 
                          outputs=gr.Label(),
-                         examples=["Normal_1.jpeg","Normal_2.jpeg","Pneumonia_1.jpeg","Pneumonia_2.jpeg"],   
+                         examples=["Normal_1.jpeg","Pneumonia_1.jpeg"],   
                          description="A simple MLP classification model for image classification using the x-ray normal lungs vs with pneumonia dataset.")
 interface.launch()