Update inference.py

inference.py

@@ -145,9 +145,9 @@ def main(args):
 
     # Get map details
     print("Getting map details.")
-    map_name = h5_image.get_maps()
-    print(f"Map Name: {map_name[0]}")
-    all_map_legends = h5_image.get_layers(map_name[0])
+    map_name = h5_image.get_maps()[0]
+    print(f"Map Name: {map_name}")
+    all_map_legends = h5_image.get_layers(map_name)
     print(f"All Map Legends: {all_map_legends}")
     
     # Filter the legends based on the feature type