Update app.py

app.py

@@ -228,6 +228,10 @@ 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))
     input_array = tf.keras.utils.img_to_array(input_image) / 255.0