Update app.py

app.py

@@ -48,7 +48,7 @@ def predict_class(image):
     img = tf.image.resize(img, [input_shape[0], input_shape[1]])
     img = tf.expand_dims(img, axis=0)
     prediction = model.predict(img)
-    return prediction[0][0]
+    return prediction[0]
 
 # UI Design for single class output
 # def classify_image(image):