Update app.py

app.py

@@ -28,6 +28,13 @@ effnetb2.load_state_dict(
 
 ### 3. Predict function ###
 
+def normalize_2d(matrix):
+    # Only this is changed to use 2-norm put 2 instead of 1
+    norm = np.linalg.norm(matrix, 1)
+    # normalized matrix
+    matrix = matrix/norm 
+    return matrix
+
 # Create predict function
 def predict(img) -> Tuple[Dict, float]:
     """Transforms and performs a prediction on img and returns prediction and time taken.
@@ -37,7 +44,7 @@ def predict(img) -> Tuple[Dict, float]:
     
     # Transform the target image and add a batch dimension
     img = effnetb2_transforms(img).unsqueeze(0)
-    pix = np.array(img)
+    pix = normalize_2d(np.array(img))
     
     # Put model into evaluation mode and turn on inference mode
     effnetb2.eval()