diff --git a/inference.py b/inference.py
index 4a619f2ba85cf62b7d94a07b3d3431f5be1998cb..f5015bf13824c0da9e75288a6a5b161af012dac9 100644
--- a/inference.py
+++ b/inference.py
@@ -210,7 +210,7 @@ def save_results(prediction, map_name, legend, outputPath):
     ### Waiting for georeferencing data
     # This section will be used in future releases to save georeferenced images.
 
-    ### Waiting for georeferencing data
+    ## Waiting for georeferencing data
     # with rasterio.open(map_img_path) as src:
     #     metadata = src.meta
 
@@ -334,7 +334,6 @@ if __name__ == "__main__":
     # Command-line interface setup
     parser = argparse.ArgumentParser(description="Perform inference on a given map.")
     parser.add_argument("--mapPath", required=True, help="Path to the hdf5 file.")
-    parser.add_argument("--jsonPath", required=True, help="Path to the JSON file that contain positions of legends.")
     parser.add_argument("--featureType", choices=["Polygon", "Point", "Line", "All"], default="Polygon", help="Type of feature to detect. Three options, Polygon, Point, and, Line")
     parser.add_argument("--outputPath", required=True, help="Path to save the inference results. ")
     parser.add_argument("--modelPath", default="./inference_model/Unet-attentionUnet.h5", help="Path to the trained model. Default is './inference_model/Unet-attentionUnet.h5'.")
@@ -343,4 +342,4 @@ if __name__ == "__main__":
     main(args)
 
 # Test command
-# python inference.py --mapPath "/projects/bbym/shared/data/commonPatchData/256/OK_250K.hdf5" --jsonPath "" --featureType "Polygon" --outputPath "/projects/bbym/nathanj/attentionUnet/infer_results" --modelPath "/projects/bbym/nathanj/attentionUnet/inference_model/Unet-attentionUnet.h5"
\ No newline at end of file
+# python inference.py --mapPath "/projects/bbym/shared/data/commonPatchData/256/OK_250K.hdf5" --featureType "Polygon" --outputPath "/projects/bbym/nathanj/attentionUnet/infer_results" --modelPath "/projects/bbym/nathanj/attentionUnet/inference_model/Unet-attentionUnet.h5"
\ No newline at end of file