Update app.py

app.py

@@ -26,6 +26,21 @@ device="cuda" if torch.cuda.is_available() else "cpu"
 
 pipe.to(device)
 
+def get_files(file_paths):
+    last_files = {}  # Dictionary to store the last file for each path
+
+    for file_path in file_paths:
+        # Split the file path into directory and file components
+        directory, file_name = file_path.rsplit('/', 1)
+    
+        # Update the last file for the current path
+        last_files[directory] = file_name
+    
+    # Extract the last files from the dictionary
+    result = list(last_files.values())
+
+    return result
+
 def load_model(custom_model):
 
     if custom_model == "":