Update app.py

app.py

@@ -26,8 +26,8 @@ def predict(image):
     
     # Get the predicted labels
     vgg_label = np.argmax(vgg_pred,axis=1)
-    mobilenet_label = np.argmax(mobilenet_pred,axis=1)[0]
-    custom_cnn_label = np.argmax(custom_cnn_pred,axis=1)[0]
+    mobilenet_label = np.argmax(mobilenet_pred,axis=1)
+    custom_cnn_label = np.argmax(custom_cnn_pred,axis=1)
     
     label_map = {0: 'Black Sea Sprat', 1: 'Gilt-Head Bream', 2: 'Hourse Mackerel', 
              3: 'Red Mullet', 4: 'Red Sea Bream', 5: 'Sea Bass',