Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -70,7 +70,10 @@ DEVICE = "cuda"
|
|
| 70 |
# 2. Empty cache for our heavy objects
|
| 71 |
_model_cache = {}
|
| 72 |
|
| 73 |
-
|
|
|
|
|
|
|
|
|
|
| 74 |
# Utility to select first image from a folder
|
| 75 |
def first_image_from_dir(directory):
|
| 76 |
patterns = ["*.jpg", "*.png", "*.jpeg"]
|
|
|
|
| 70 |
# 2. Empty cache for our heavy objects
|
| 71 |
_model_cache = {}
|
| 72 |
|
| 73 |
+
def first_file_from_dir(directory, ext):
|
| 74 |
+
files = glob.glob(os.path.join(directory, f"*.{ext}"))
|
| 75 |
+
return sorted(files)[0] if files else None#
|
| 76 |
+
|
| 77 |
# Utility to select first image from a folder
|
| 78 |
def first_image_from_dir(directory):
|
| 79 |
patterns = ["*.jpg", "*.png", "*.jpeg"]
|