Update app.py

app.py

@@ -86,21 +86,13 @@ VIDEO_EXTENSIONS = {'.mp4', '.mkv', '.mov', '.avi', '.flv', '.wmv', '.webm', '.m
 
 
 def get_file_extension(filename):
-    return os.path.splitext(filename)[1]
-
-
+    return os.path.splitext(filename)[1].lower()
 
 def is_image(filename):
-    image_extensions = ['.png', '.jpg', '.jpeg', '.gif', '.bmp', '.tiff']
-    ext = os.path.splitext(filename)[1].lower()
-    return ext in image_extensions
-
-import os
+    return get_file_extension(filename) in IMAGE_EXTENSIONS
 
 def is_video(filename):
-    video_extensions = ['.mp4', '.avi', '.mov', '.mkv', '.flv', '.wmv']
-    ext = os.path.splitext(filename)[1].lower()
-    return ext in video_extensions
+    return get_file_extension(filename) in VIDEO_EXTENSIONS
 
 
 def load_model_and_tokenizer():