downloadable files

app.py

@@ -37,14 +37,6 @@ load_img = LoadImage(image_only=True)
 #SET CLASSES
 class_names = CLASSES
 
-#SET IMAGE PATH LIST FOR STREAMLIT'S SELECT BOX
-filelist=[""]
-for root, dirs, files in os.walk("images/raw"):
-      for file in files:
-             filename=file.split(".")[0]
-             filelist.append(filename)
-filelist = tuple(filelist)
-
 #SILENCE STREAMIT WARNING
 st.set_option('deprecation.showPyplotGlobalUse', False)
 
@@ -73,7 +65,7 @@ with st.sidebar:
     st.title("Alzheimer Classifier Demo")
     img_path = st.selectbox(
             "Select Image",
-            filelist,
+            class_names,
             on_change= click_false,
         )
     col1, col2 = st.columns((1,1))