Update app.py

app.py

@@ -38,7 +38,8 @@ 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.uint8)
+    output = output.astype(np.float64)
+    img = img.astype(np.float64)
 
     overlayed_image = cv2.addWeighted(img,0.4,output,0.1,0)