Update app.py

app.py

@@ -41,6 +41,8 @@ def visualize_model_output(prediction, img):
     output = output.astype(np.float64)
     img = img.astype(np.float64)
 
+    img = resize_image(img, output.shape[0], output.shape[1])
+
     print(output.shape, img.shape,'shapes')
     
     added_image = cv2.addWeighted(img,1.,output,1.,0)