Update app.py

app.py

@@ -72,7 +72,9 @@ with st.sidebar:
 st.write("サイドバーより学習データをZipファイルとしてアップロードしボタンをクリック.")
 if st.button("訓練開始"):
     with st.spinner("1分ほどお待ちください..."):
-        image_paths = glob.glob("data/*/*.JPG")
+        image_paths = []
+        for ext in ["JPG", "jpg", "PNG", "png"]:
+            image_paths.extend(glob.glob(f"data/*/*.{ext}"))
         col1, col2, col3 = st.columns(3)  # 2列のコンテナを用意する
         with col1:
             st.image(image_paths[0])