Update app.py

app.py

@@ -94,7 +94,12 @@ def plot_outlines(img, masks):
     return pil_img
 
 def plot_overlay(img, masks):
-    img = normalize99(img.astype(np.float32).mean(axis=-1))
+    if img.ndim>2:
+        img_gray = img.astype(np.float32).mean(axis=-1)
+    else:
+        img_gray = img.astype(np.float32)
+        
+    img = normalize99(img_gray)
     img -= img.min()
     img /= img.max()
     HSV = np.zeros((img.shape[0], img.shape[1], 3), np.float32)