model_prediction

app.py

@@ -30,10 +30,7 @@ def model_prediction(image, model):
     return result
 
 def model_prediction(image, model):
-    img = cv2.imread(image)
-    plt.imshow(img)
-    plt.show()
-    resize = tf.image.resize(img, (256,256))
+    resize = tf.image.resize(image, (256,256))
     yhat = model.predict(np.expand_dims(resize/255, 0))
     if(yhat>0.5):
         result = "Prediction is loose"