Update app.py

app.py

@@ -107,7 +107,7 @@ def highlight_segmentation(x, y):
         try:
             seg_value = seg.numpy()[y, x]
             print(f"Mouse over position: ({x}, {y}), Segmentation value: {seg_value}")
-        except IndexError:
+        except (AttributeError, IndexError):
             print(f"Invalid mouse position: ({x}, {y}), Segmentation value cannot be retrieved.")
     else:
         print("Segmentation information is not available.")