Update app.py

app.py

@@ -38,10 +38,12 @@ 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.int16)
-    img = img.astype(np.int16)
+    output = output.astype(np.float64)
+    img = img.astype(np.float64)
+
+    print(output.shape, img.shape,'shapes')
     
-    added_image = cv2.addWeighted(img,1,output,1,0)
+    added_image = cv2.addWeighted(img,1.,output,1.,0)
 
     return added_image