Fixed sample image path issues

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import random
+import pathlib
 import numpy as np
 from PIL import Image
 import torch
@@ -39,16 +40,23 @@ def read_image(path):
     return data
 
 
+# def sample_images():
+#     images = []
+#     length = len(datamodule.test_dataset)
+#     classes = datamodule.train_dataset.classes
+#     for i in range(10):
+#         idx = random.randint(0, length - 1)
+#         image, label = datamodule.test_dataset[idx]
+#         image = inv_normalize(image).permute(1, 2, 0).numpy()
+#         images.append((image, classes[label]))
+#     return images
+
+
 def sample_images():
-    images = []
-    length = len(datamodule.test_dataset)
-    classes = datamodule.train_dataset.classes
-    for i in range(10):
-        idx = random.randint(0, length - 1)
-        image, label = datamodule.test_dataset[idx]
-        image = inv_normalize(image).permute(1, 2, 0).numpy()
-        images.append((image, classes[label]))
-    return images
+    sample_imges_dir = pathlib.Path("./sample_images")
+    sample_images = list(sample_imges_dir.iterdir())
+    sample_image_labels = [image.stem for image in sample_images]
+    return list(zip(sample_images, sample_image_labels))
 
 
 def get_misclassified_images(misclassified_count):