add condition

app.py

@@ -29,7 +29,7 @@ def get_list_of_images():
     # return fs.ls("datasets/aifred-smart-life-coach/capstone-images", detail=False, refresh=True)
     api = hfh.HfApi()
     files = api.list_repo_tree(repo_id="aifred-smart-life-coach/capstone-images", repo_type="dataset", recursive=True,)
-    files = [file.path for file in files]
+    files = [file.path for file in files if file.path.endswith((".png", ".jpg"))]
     return files