Update app.py

app.py

@@ -19,7 +19,19 @@ shutil.move("HairFastGAN/input", "input")
 
 shutil.rmtree("HairFastGAN")
 
-from hair_swap import HairFast, get_parser
+directory_path = 'pretrained_models'
+# Iterate over each entry in the directory
+for entry in os.listdir(directory_path):
+    # Create full path to the entry
+    full_path = os.path.join(directory_path, entry)
+    # Check if it's a file
+    if os.path.isfile(full_path):
+        # Get the file size
+        size = os.path.getsize(full_path)
+        # Print file name and its size in bytes
+        print(f"{entry}: {size} bytes")
+
+from HairFastGAN.hair_swap import HairFast, get_parser
 
 hair_fast = HairFast(get_parser().parse_args([]))