Spaces:
Running
on
Zero
Running
on
Zero
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"]
|