Update app.py

app.py

@@ -46,8 +46,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)
-    print(prediction[0])
-    return prediction
+    return prediction[0]
 
 # UI Design
 # def classify_image(image):