Update app.py

app.py

@@ -38,11 +38,13 @@ def visualize_model_output(prediction, img):
         output[:,:,1][prediction[:,:,0]==unq_class] = rgb_class_unique[1]
         output[:,:,2][prediction[:,:,0]==unq_class] = rgb_class_unique[2]
 
-    output = output.astype(np.float64)
-    img = img.astype(np.float64)
+
 
     img = resize_image(img, output.shape[0], output.shape[1])
 
+    output = output.astype(np.int32)
+    img = img.astype(np.int32)
+
     print(output.shape, img.shape,'shapes')
     
     added_image = cv2.addWeighted(img,1.,output,1.,0)