Update app.py

app.py

@@ -39,15 +39,10 @@ def visualize_model_output(prediction, img):
         output[:,:,2][prediction[:,:,0]==unq_class] = rgb_class_unique[2]
 
     output = output.astype(np.uint8)
-    output_color_coverted = cv2.cvtColor(output, cv2.COLOR_BGR2RGB) 
-    im_pil_output = Image.fromarray(output_color_coverted)
 
-    color_coverted = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) 
-    im_pil = Image.fromarray(color_coverted)
+    overlayed_image = cv2.addWeighted(img,0.4,output,0.1,0)
 
-    im_pil.paste(im_pil_output, (0,0))
-
-    return im_pil.astype(np.uint8)
+    return overlayed_image.astype(np.uint8)
 
 
 def do_prediction(model_name, img):