Update app.py

app.py

@@ -36,16 +36,16 @@ model.eval()
 
 EXAMPLES_FOLDER = 'examples'
 examples_names = os.listdir(EXAMPLES_FOLDER)
-# examples = []
-# for example_name in examples_names:
-#     example_path = os.path.join(EXAMPLES_FOLDER, example_name)
-#     label = example_name.split('_')[0]
-#     example = {
-#         'path': example_path,
-#         'label': label
-#     }
-#     examples.append(example)
-# np.random.shuffle(examples) # shuffle
+examples = []
+for example_name in examples_names:
+    example_path = os.path.join(EXAMPLES_FOLDER, example_name)
+    label = example_name.split('_')[0]
+    example = {
+        'path': example_path,
+        'label': label
+    }
+    examples.append(example)
+np.random.shuffle(examples) # shuffle
 
 def predict(input_image:Image.Image, true_label:str):
     """Predict the label of the input_image"""