Update app.py

app.py

@@ -134,23 +134,23 @@ def load_data():
 
     imgs = y_test["Path"].values
     labels = y_test["ClassId"].values
-
-    test_images = []
-    for img in imgs:
-        if isinstance(img,str):
-            image = Image.open('traffic_lights/Test/'+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")