Update app.py

app.py

@@ -134,23 +134,23 @@ def load_data():
 
     imgs = y_test["Path"].values
     labels = y_test["ClassId"].values
-    st.write(imgs)
-    # test_images = []
-    # for img in imgs:
-    #     if isinstance(img,str):
-    #         image = Image.open('traffic_lights/'+img)
-    #         image = image.resize([30, 30])
-    #         test_images.append(np.array(image))
-
-    # # Load meta images
-    # meta_images = {}
-    # meta_folder = 'traffic_lights/Meta'  # Replace with the path to your meta folder
-    # for class_id in range(43):
-    #     meta_image_path = os.path.join(meta_folder, f"{class_id}.png")  # Assuming meta images are named as 0.png, 1.png, etc.
-    #     if os.path.exists(meta_image_path):
-    #         meta_images[class_id] = Image.open(meta_image_path)
-
-    # return test_images, labels, meta_images
+    # st.write(imgs)
+    test_images = []
+    for img in imgs:
+        if isinstance(img,str):
+            image = Image.open('traffic_lights'+img)
+            image = image.resize([30, 30])
+            test_images.append(np.array(image))
+
+    # Load meta images
+    meta_images = {}
+    meta_folder = 'traffic_lights/Meta'  # Replace with the path to your meta folder
+    for class_id in range(43):
+        meta_image_path = os.path.join(meta_folder, f"{class_id}.png")  # Assuming meta images are named as 0.png, 1.png, etc.
+        if os.path.exists(meta_image_path):
+            meta_images[class_id] = Image.open(meta_image_path)
+
+    return test_images, labels, meta_images
 
 def main():
     colored_title("Traffic Symbol Prediction", "black")