debug: removed labels

yolov8.py

@@ -160,10 +160,10 @@ def dff_nmf(image, target_lyr, n_components):
     concepts, batch_explanations, explanations = dff(input_tensor, model, n_components)
 
 
-    yolov5_categories_url = \
+    #yolov5_categories_url = \
             "https://github.com/ultralytics/yolov5/raw/master/data/coco128.yaml"  # URL to the YOLOv5 categories file
-    yaml_data = requests.get(yolov5_categories_url).text
-    labels = yaml.safe_load(yaml_data)['names']  # Parse the YAML file to get class names
+    #yaml_data = requests.get(yolov5_categories_url).text
+   # labels = yaml.safe_load(yaml_data)['names']  # Parse the YAML file to get class names
     num_classes = model.model.model[-1].nc 
     results = []
     for indx in range(explanations[0].shape[0]):
@@ -200,8 +200,8 @@ def dff_nmf(image, target_lyr, n_components):
                 linewidth=2, edgecolor='r', facecolor='none')
         ax.add_patch(rect)
     
-        predicted_label = labels[top_class_id]  # Map ID to label
-        ax.text(x1, y1, f"{predicted_label}: {top_score:.2f}",
+        #predicted_label = labels[top_class_id]  # Map ID to label
+        #ax.text(x1, y1, f"{predicted_label}: {top_score:.2f}",
             color='r', fontsize=12, verticalalignment='top')
         plt.subplots_adjust(left=0, right=1, top=1, bottom=0)