Update app.py

app.py

@@ -235,10 +235,10 @@ def classifyImage(input_image):
     # Add a batch dimension 
     input_array = tf.expand_dims(input_array, 0)  # (1, 224, 224, 3)
 
-    temp = tf.nn.softmax(predictions).numpy()
-    print("SOFTMAX PREDICTIONS",temp)
     predictions = model.predict(input_array)[0]
-    print(f"Predictions: {predictions}")
+    print(f"Predictions BEFORE SOFTMAX: {predictions}")
+    predictions = tf.nn.softmax(predictions).numpy()
+    print(f"Predictions AFTER SOFTMAX: {predictions}")
 
     # Sort predictions to get top 5
     top_indices = np.argsort(predictions)[-5:][::-1]