Update app.py

app.py

@@ -38,10 +38,10 @@ 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.int8)
-    img = img.astype(np.int8)
+    output = output.astype(np.int16)
+    img = img.astype(np.int16)
     
-    added_image = cv2.addWeighted(img,0.4,output,0.1,0)
+    added_image = cv2.addWeighted(img,1,output,1,0)
 
     return added_image