alexnasa commited on
Commit
1d930bb
·
verified ·
1 Parent(s): dfdcbbc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
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"]