Update app.py

app.py

@@ -1363,7 +1363,8 @@ def display_images_and_wikipedia_summaries(num_columns=4):
 
 def display_videos_and_links(num_columns):
     #video_files = [f for f in os.listdir('.') if f.endswith('.mp4')]
-    video_files = [f for f in os.listdir('.') if f.endswith('.webm')]
+    #video_files = [f for f in os.listdir('.') if f.endswith('.webm')]
+    video_files = [f for f in os.listdir('.') if f.endswith(('.mp4', '.webm'))]
 
     if not video_files:
         st.write("No MP4 videos found in the current directory.")