Update app.py

app.py

@@ -41,15 +41,9 @@ def visualize_model_output(prediction, img):
     output = output.astype(np.int8)
     img = img.astype(np.int8)
     
-    bg_img = img.copy()
-    shapes = np.zeros_like(img, np.uint8)
-    shapes = output
-    mask = shapes.astype(bool)
-    alpha = 0.5
-    bg_img[mask] = cv2.addWeighted(bg_img, 1 - alpha, shapes, alpha, 0)[mask]
-    
+    added_image = cv2.addWeighted(img,0.4,output,0.1,0)
 
-    return bg_img
+    return added_image
 
 
 def do_prediction(model_name, img):