Update visualization.py

visualization.py

@@ -110,7 +110,7 @@ def plot_from_reranker_output(reranker_output, draw_box=True, draw_arcs=True):
                 Image.open(reranker_output["chm-nearest-neighbors"][i]).convert("RGB")
             )
         )
-        axes[0][1 + i].set_title(f"CHM - Top - {i+1}", fontsize=21)
+        axes[0][1 + i].set_title(f"CHM-Corr - Top - {i+1}", fontsize=21)
 
     if reranker_output["chm-prediction-confidence"] < 5:
         for i in range(reranker_output["chm-prediction-confidence"], 5):
@@ -255,7 +255,7 @@ def plot_from_reranker_output(reranker_output, draw_box=True, draw_arcs=True):
     fig.text(
         0.5,
         0.95,
-        f"CHM: {reranker_output['chm-prediction']}",
+        f"CHM-Corr Prediction: {reranker_output['chm-prediction']}",
         ha="center",
         va="bottom",
         color="black",