new

pages/Model_Evaluation.py

@@ -61,15 +61,20 @@ def apply_gaussian_filter(image, kernel_size=(5, 5), sigma=1.0):
 
 # ---- Custom Dataset ----
 def __getitem__(self, idx):
-    img_url = self.image_paths[idx]
+    # Get the image URL and label
+    raw_url = self.image_paths[idx]
     label = int(self.labels[idx])
 
+    # Fix the URL if it's in blob format
+    img_url = raw_url.replace("/blob/", "/resolve/")
+
     try:
         print(f"Downloading: {img_url}")
-        response = requests.get(img_url)
+        response = requests.get(img_url, stream=True)
         if response.status_code != 200:
             raise ValueError(f"Failed to download image from {img_url} - status code: {response.status_code}")
 
+        # Read and convert the image
         image = Image.open(BytesIO(response.content)).convert("RGB")
     except Exception as e:
         raise RuntimeError(f"Error loading image from {img_url}: {e}")