push

app.py

@@ -72,13 +72,13 @@ def inference(input_img, transparency = 0.5, target_layer_number = -1, top_predi
     top_n_confidences = dict(list(sorted_confidences.items())[:top_predictions])
 
 
-    files = os.listdir('/content/misclassified_images/')
+    files = os.listdir('misclassified_images/')
 
     # Plot the misclassified images
     fig = plt.figure(figsize=(12, 5))
     for i in range(miss_classified_images_count):
         sub = fig.add_subplot(2, 5, i+1)
-        npimg = Image.open('/content/misclassified_images/' + files[i])
+        npimg = Image.open('misclassified_images/' + files[i])
 
         # Use regex to extract target and predicted classes
         match = re.search(r'Target_(\w+)_Pred_(\w+)_\d+.jpeg', files[i])