Limiting example images displayed to 5

app.py

@@ -240,9 +240,14 @@ def run_diagnosis(backbone_name, input_image, preprocess_fn = None, Idx2labels=N
 example_list_dir = os.path.join(os.getcwd(), "Curated test samples")
 example_list_img_names = os.listdir(example_list_dir)
 
+# example_list = [
+#     [os.path.join(example_list_dir, example_img), configs["DEFAULT_BACKBONE"]]
+#     for example_img in example_list_img_names
+#     if example_img.lower().endswith(".png")]
+
 example_list = [
-    [os.path.join(example_list_dir, example_img), configs["DEFAULT_BACKBONE"]]
-    for example_img in example_list_img_names
+    [configs["DEFAULT_BACKBONE"], os.path.join(example_list_dir, example_img)]
+    for example_img in example_list_img_names[:6]
     if example_img.lower().endswith(".png")]
 
 # example_list = [['/content/new_labels.csv',"ResNet50"]]