Update app.py

app.py

@@ -20,21 +20,16 @@ shutil.move("HairFastGAN/input", "input")
 
 shutil.rmtree("HairFastGAN")
 
-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")
-
-hairfastgan_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'HairFastGAN')
-print(hairfastgan_path)
-sys.path.append(hairfastgan_path)
+items = os.listdir()
+
+for item in items:
+    print(item)
+    shutil.move(item, os.path.join('..', item))
+
+os.chdir("..")
+
+shutil.rmtree("HairFastGAN")
+
 from hair_swap import HairFast, get_parser
 
 hair_fast = HairFast(get_parser().parse_args([]))