Update app.py

app.py

@@ -228,9 +228,6 @@ class_names = [
     "tuna_tartare", "waffles"
 ]
 
-def softmax_with_temperature(logits, temperature=1.0):
-    logits = logits / temperature
-    return tf.nn.softmax(logits)
     
 def classifyImage(input_image):
     input_image = input_image.resize((img_size, img_size))
@@ -241,11 +238,17 @@ def classifyImage(input_image):
 
     predictions = model.predict(input_array)[0]
     print(f"Predictions: {predictions}")
-    predictions = softmax_with_temperature(predictions, temperature=2.0).numpy() #testing
+
+        
+    probability_sum = predictions.sum() * 100
+    print(f"Sum of predictions BEFORE SOFTMAX(as percentages): {probability_sum:.2f}%")
+
+    
+    predictions = tf.nn.softmax(predictions).numpy() #testing
     print(f"Predictions AFTER SOFTMAX: {predictions}")
     
     probability_sum = predictions.sum() * 100
-    print(f"Sum of predictions (as percentages): {probability_sum:.2f}%")
+    print(f"Sum of predictions AFTER SOFTMAX(as percentages): {probability_sum:.2f}%")
     
     for i, confidence in enumerate(predictions):
         print(f"Class {i} has {confidence*100:.2f}% confidence")