Update app.py

Updated to only 2 classes

app.py

@@ -6,7 +6,7 @@ import pandas as pd
 import matplotlib.pyplot as plt
 
 # Load the trained model
-model_path = "skin-model_transferlearning.keras"
+model_path = "skin-model_transferlearning4.keras"
 model = tf.keras.models.load_model(model_path)
 
 # Define the core prediction function
@@ -24,7 +24,7 @@ def predict_skin(image):
     probabilities = tf.nn.softmax(prediction, axis=1)
     
     # Map probabilities to Skin condition classes
-    class_names = ['AKIEC', 'BCC', 'BKL', 'DF', 'MEL', 'NV', 'VASC']
+    class_names = ['BKL', 'MAL']
     probabilities_dict = {skin_class: round(float(probability), 2) for skin_class, probability in zip(class_names, probabilities.numpy()[0])}
     
     return probabilities_dict