assentian1970 commited on
Commit
ae25eea
·
verified ·
1 Parent(s): 8a2fb86

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -93,6 +93,12 @@ def is_image(filename):
93
  ext = os.path.splitext(filename)[1].lower()
94
  return ext in image_extensions
95
 
 
 
 
 
 
 
96
 
97
 
98
  def load_model_and_tokenizer():
 
93
  ext = os.path.splitext(filename)[1].lower()
94
  return ext in image_extensions
95
 
96
+ import os
97
+
98
+ def is_video(filename):
99
+ video_extensions = ['.mp4', '.avi', '.mov', '.mkv', '.flv', '.wmv']
100
+ ext = os.path.splitext(filename)[1].lower()
101
+ return ext in video_extensions
102
 
103
 
104
  def load_model_and_tokenizer():