Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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():
|