Update app.py

app.py

@@ -93,6 +93,15 @@ def load_model():
 
     return model, processor, device
 
+def show_mask(mask, ax, random_color=False):
+    if random_color:
+        color = np.concatenate([np.random.random(3), np.array([0.6])], axis=0)
+    else:
+        color = np.array([30/255, 144/255, 255/255, 0.6])
+    h, w = mask.shape[-2:]
+    mask_image = mask.reshape(h, w, 1) * color.reshape(1, 1, -1)
+    ax.imshow(mask_image)
+
 ### SERVER ###
 def server(input: Inputs, output: Outputs, session: Session):