Fix: comment irrelevant codebase

yolov5.py

@@ -82,7 +82,7 @@ def xai_yolov5(image):
     return Image.fromarray(final_image), caption
 
 
- 
+"""
 import yaml
 import torch
 import warnings
@@ -104,7 +104,7 @@ std = [0.229, 0.224, 0.225]   # Standard deviation for RGB channels
 model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True).to(device)
 print(f"Loaded YOLOv5 model on {device}")
 def create_labels(concept_scores, top_k=2):
-    """Create a list with the category names of the top scoring categories."""
+    
     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
@@ -123,9 +123,7 @@ def create_labels(concept_scores, top_k=2):
     return concept_labels_topk
 
 def get_image_from_url(url, device):
-    """A function that gets a URL of an image,
-    and returns a numpy image and a preprocessed
-    torch tensor ready to pass to the model"""
+ 
 
     img = np.array(Image.open(os.path.join(os.getcwd(), "data/xai/sample1.jpeg")))
     img = cv2.resize(img, (640, 640))
@@ -170,3 +168,4 @@ def visualize_image(model, img_url, n_components=20, top_k=1, lyr_idx = 2):
 for indx in range(2,12):
     Image.fromarray(visualize_image(model, 
                                         "https://github.com/jacobgil/pytorch-grad-cam/blob/master/examples/both.png?raw=true", lyr_idx = indx))
+"""