Update make_predictions.py

make_predictions.py

@@ -8,7 +8,7 @@ def make_predictions(model, input_img, threshold = 0.05):
         image = input_img
         image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
         image = image.astype("float32") / 255.0
-        image = cv2.resize(image, (height, width))
+        image = cv2.resize(image, (256, 256))
         image = np.transpose(image, (2, 0, 1))
         image = np.expand_dims(image, 0)
         image = torch.from_numpy(image).to(device)