Update app.py

app.py

@@ -18,6 +18,8 @@ import matplotlib.pyplot as plt
 from matplotlib.colors import LinearSegmentedColormap
 import textwrap
 import plotly.graph_objects as go
+from tensorflow.keras.applications.efficientnet import preprocess_input
+
 
 st.set_page_config(
     page_title="Food Chain", 
@@ -234,10 +236,11 @@ def classifyImage(input_image):
 
     # Add a batch dimension 
     input_array = tf.expand_dims(input_array, 0)  # (1, 224, 224, 3)
+    input_array = preprocess_input(input_array) #TESTING
 
     predictions = model.predict(input_array)[0]
     print(f"Predictions: {predictions}")
-    predictions = tf.nn.softmax(predictions).numpy()
+    predictions = tf.nn.softmax(predictions).numpy() #TESTING
     print(f"Predictions AFTER SOFTMAX: {predictions}")
 
     # Sort predictions to get top 5