diff --git a/GanAttack.py b/GanAttack.py
index 700c760..e2f5944 100644
--- a/GanAttack.py
+++ b/GanAttack.py
@@ -1,3 +1,5 @@
+import sys
+sys.path.append('./GanInverter')
 import torch
 import torch.nn as nn
 import torch.optim as optim
@@ -5,8 +7,10 @@ from torchvision import models
 from omegaconf import DictConfig, OmegaConf
 from data.dataset import get_dataset,get_adv_dataset
 from utils import get_model,set_requires_grad
-from model.GanInverter.models.stylegan2.model import Generator
-from model.GanInverter.inference.two_stage_inference import TwoStageInference
+from GanInverter.inference.two_stage_inference import TwoStageInference
+from GanInverter.models.stylegan2.model import Generator
+# from models import GanAttack
+import model
 from model import GanAttack,CLIPLoss,VggLoss,get_prompt
 import torch.nn.functional as F
 import time
@@ -16,6 +20,8 @@ import hydra
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
 
+
+
 def get_stylegan_generator(cfg):
 
     # ensure_checkpoint_exists(ckpt_path)
diff --git a/model/GanInverter/.gitignore b/GanInverter/.gitignore
similarity index 100%
rename from model/GanInverter/.gitignore
rename to GanInverter/.gitignore
diff --git a/model/GanInverter/LICENSE b/GanInverter/LICENSE
similarity index 100%
rename from model/GanInverter/LICENSE
rename to GanInverter/LICENSE
diff --git a/model/GanInverter/README.md b/GanInverter/README.md
similarity index 100%
rename from model/GanInverter/README.md
rename to GanInverter/README.md
diff --git a/model/GanInverter/configs/__init__.py b/GanInverter/configs/__init__.py
similarity index 100%
rename from model/GanInverter/configs/__init__.py
rename to GanInverter/configs/__init__.py
diff --git a/model/GanInverter/configs/dhr/dhr.yaml b/GanInverter/configs/dhr/dhr.yaml
similarity index 100%
rename from model/GanInverter/configs/dhr/dhr.yaml
rename to GanInverter/configs/dhr/dhr.yaml
diff --git a/model/GanInverter/configs/dhr/readme.md b/GanInverter/configs/dhr/readme.md
similarity index 100%
rename from model/GanInverter/configs/dhr/readme.md
rename to GanInverter/configs/dhr/readme.md
diff --git a/model/GanInverter/configs/e4e/e4e_ffhq_r50.yaml b/GanInverter/configs/e4e/e4e_ffhq_r50.yaml
similarity index 100%
rename from model/GanInverter/configs/e4e/e4e_ffhq_r50.yaml
rename to GanInverter/configs/e4e/e4e_ffhq_r50.yaml
diff --git a/model/GanInverter/configs/lsap/lsap_ffhq_r50.yaml b/GanInverter/configs/lsap/lsap_ffhq_r50.yaml
similarity index 100%
rename from model/GanInverter/configs/lsap/lsap_ffhq_r50.yaml
rename to GanInverter/configs/lsap/lsap_ffhq_r50.yaml
diff --git a/model/GanInverter/configs/lsap/readme.md b/GanInverter/configs/lsap/readme.md
similarity index 100%
rename from model/GanInverter/configs/lsap/readme.md
rename to GanInverter/configs/lsap/readme.md
diff --git a/model/GanInverter/configs/optim/optim_celeba-hq.yaml b/GanInverter/configs/optim/optim_celeba-hq.yaml
similarity index 100%
rename from model/GanInverter/configs/optim/optim_celeba-hq.yaml
rename to GanInverter/configs/optim/optim_celeba-hq.yaml
diff --git a/model/GanInverter/configs/paths_config.py b/GanInverter/configs/paths_config.py
similarity index 100%
rename from model/GanInverter/configs/paths_config.py
rename to GanInverter/configs/paths_config.py
diff --git a/model/GanInverter/configs/psp/psp_ffhq_r50.yaml b/GanInverter/configs/psp/psp_ffhq_r50.yaml
similarity index 100%
rename from model/GanInverter/configs/psp/psp_ffhq_r50.yaml
rename to GanInverter/configs/psp/psp_ffhq_r50.yaml
diff --git a/model/GanInverter/configs/pti/pti.yaml b/GanInverter/configs/pti/pti.yaml
similarity index 100%
rename from model/GanInverter/configs/pti/pti.yaml
rename to GanInverter/configs/pti/pti.yaml
diff --git a/model/GanInverter/configs/pti/pti_pivot.yaml b/GanInverter/configs/pti/pti_pivot.yaml
similarity index 100%
rename from model/GanInverter/configs/pti/pti_pivot.yaml
rename to GanInverter/configs/pti/pti_pivot.yaml
diff --git a/model/GanInverter/configs/sam/sam.yaml b/GanInverter/configs/sam/sam.yaml
similarity index 100%
rename from model/GanInverter/configs/sam/sam.yaml
rename to GanInverter/configs/sam/sam.yaml
diff --git a/model/GanInverter/configs/transforms_config.py b/GanInverter/configs/transforms_config.py
similarity index 100%
rename from model/GanInverter/configs/transforms_config.py
rename to GanInverter/configs/transforms_config.py
diff --git a/model/GanInverter/criteria/__init__.py b/GanInverter/criteria/__init__.py
similarity index 100%
rename from model/GanInverter/criteria/__init__.py
rename to GanInverter/criteria/__init__.py
diff --git a/model/GanInverter/criteria/id_loss.py b/GanInverter/criteria/id_loss.py
similarity index 100%
rename from model/GanInverter/criteria/id_loss.py
rename to GanInverter/criteria/id_loss.py
diff --git a/model/GanInverter/criteria/lpips/__init__.py b/GanInverter/criteria/lpips/__init__.py
similarity index 100%
rename from model/GanInverter/criteria/lpips/__init__.py
rename to GanInverter/criteria/lpips/__init__.py
diff --git a/model/GanInverter/criteria/lpips/lpips.py b/GanInverter/criteria/lpips/lpips.py
similarity index 100%
rename from model/GanInverter/criteria/lpips/lpips.py
rename to GanInverter/criteria/lpips/lpips.py
diff --git a/model/GanInverter/criteria/lpips/networks.py b/GanInverter/criteria/lpips/networks.py
similarity index 100%
rename from model/GanInverter/criteria/lpips/networks.py
rename to GanInverter/criteria/lpips/networks.py
diff --git a/model/GanInverter/criteria/lpips/utils.py b/GanInverter/criteria/lpips/utils.py
similarity index 100%
rename from model/GanInverter/criteria/lpips/utils.py
rename to GanInverter/criteria/lpips/utils.py
diff --git a/model/GanInverter/criteria/moco_loss.py b/GanInverter/criteria/moco_loss.py
similarity index 100%
rename from model/GanInverter/criteria/moco_loss.py
rename to GanInverter/criteria/moco_loss.py
diff --git a/model/GanInverter/criteria/w_norm.py b/GanInverter/criteria/w_norm.py
similarity index 100%
rename from model/GanInverter/criteria/w_norm.py
rename to GanInverter/criteria/w_norm.py
diff --git a/model/GanInverter/datasets/__init__.py b/GanInverter/datasets/__init__.py
similarity index 100%
rename from model/GanInverter/datasets/__init__.py
rename to GanInverter/datasets/__init__.py
diff --git a/model/GanInverter/datasets/images_dataset.py b/GanInverter/datasets/images_dataset.py
similarity index 100%
rename from model/GanInverter/datasets/images_dataset.py
rename to GanInverter/datasets/images_dataset.py
diff --git a/model/GanInverter/datasets/inference_dataset.py b/GanInverter/datasets/inference_dataset.py
similarity index 100%
rename from model/GanInverter/datasets/inference_dataset.py
rename to GanInverter/datasets/inference_dataset.py
diff --git a/model/GanInverter/docs/dataset.md b/GanInverter/docs/dataset.md
similarity index 100%
rename from model/GanInverter/docs/dataset.md
rename to GanInverter/docs/dataset.md
diff --git a/model/GanInverter/docs/dhr.png b/GanInverter/docs/dhr.png
similarity index 100%
rename from model/GanInverter/docs/dhr.png
rename to GanInverter/docs/dhr.png
diff --git a/model/GanInverter/docs/gan_inverter.jpeg b/GanInverter/docs/gan_inverter.jpeg
similarity index 100%
rename from model/GanInverter/docs/gan_inverter.jpeg
rename to GanInverter/docs/gan_inverter.jpeg
diff --git a/model/GanInverter/docs/inference_pipeline.png b/GanInverter/docs/inference_pipeline.png
similarity index 100%
rename from model/GanInverter/docs/inference_pipeline.png
rename to GanInverter/docs/inference_pipeline.png
diff --git a/model/GanInverter/docs/install.md b/GanInverter/docs/install.md
similarity index 100%
rename from model/GanInverter/docs/install.md
rename to GanInverter/docs/install.md
diff --git a/model/GanInverter/docs/lsap.png b/GanInverter/docs/lsap.png
similarity index 100%
rename from model/GanInverter/docs/lsap.png
rename to GanInverter/docs/lsap.png
diff --git a/model/GanInverter/editing/__init__.py b/GanInverter/editing/__init__.py
similarity index 100%
rename from model/GanInverter/editing/__init__.py
rename to GanInverter/editing/__init__.py
diff --git a/model/GanInverter/editing/base_editing.py b/GanInverter/editing/base_editing.py
similarity index 100%
rename from model/GanInverter/editing/base_editing.py
rename to GanInverter/editing/base_editing.py
diff --git a/model/GanInverter/editing/ganspace.py b/GanInverter/editing/ganspace.py
similarity index 100%
rename from model/GanInverter/editing/ganspace.py
rename to GanInverter/editing/ganspace.py
diff --git a/model/GanInverter/editing/interfacegan.py b/GanInverter/editing/interfacegan.py
similarity index 100%
rename from model/GanInverter/editing/interfacegan.py
rename to GanInverter/editing/interfacegan.py
diff --git a/model/GanInverter/inference/__init__.py b/GanInverter/inference/__init__.py
similarity index 100%
rename from model/GanInverter/inference/__init__.py
rename to GanInverter/inference/__init__.py
diff --git a/model/GanInverter/inference/code_infer.py b/GanInverter/inference/code_infer.py
similarity index 100%
rename from model/GanInverter/inference/code_infer.py
rename to GanInverter/inference/code_infer.py
diff --git a/model/GanInverter/inference/dhr_infer.py b/GanInverter/inference/dhr_infer.py
similarity index 100%
rename from model/GanInverter/inference/dhr_infer.py
rename to GanInverter/inference/dhr_infer.py
diff --git a/model/GanInverter/inference/encoder_infer.py b/GanInverter/inference/encoder_infer.py
similarity index 100%
rename from model/GanInverter/inference/encoder_infer.py
rename to GanInverter/inference/encoder_infer.py
diff --git a/model/GanInverter/inference/inference.py b/GanInverter/inference/inference.py
similarity index 100%
rename from model/GanInverter/inference/inference.py
rename to GanInverter/inference/inference.py
diff --git a/model/GanInverter/inference/optim_infer.py b/GanInverter/inference/optim_infer.py
similarity index 100%
rename from model/GanInverter/inference/optim_infer.py
rename to GanInverter/inference/optim_infer.py
diff --git a/model/GanInverter/inference/pti_infer.py b/GanInverter/inference/pti_infer.py
similarity index 100%
rename from model/GanInverter/inference/pti_infer.py
rename to GanInverter/inference/pti_infer.py
diff --git a/model/GanInverter/inference/sam_infer.py b/GanInverter/inference/sam_infer.py
similarity index 100%
rename from model/GanInverter/inference/sam_infer.py
rename to GanInverter/inference/sam_infer.py
diff --git a/model/GanInverter/inference/two_stage_inference.py b/GanInverter/inference/two_stage_inference.py
similarity index 100%
rename from model/GanInverter/inference/two_stage_inference.py
rename to GanInverter/inference/two_stage_inference.py
diff --git a/model/GanInverter/licenses/LICENSE_HuangYG123 b/GanInverter/licenses/LICENSE_HuangYG123
similarity index 100%
rename from model/GanInverter/licenses/LICENSE_HuangYG123
rename to GanInverter/licenses/LICENSE_HuangYG123
diff --git a/model/GanInverter/licenses/LICENSE_S-aiueo32 b/GanInverter/licenses/LICENSE_S-aiueo32
similarity index 100%
rename from model/GanInverter/licenses/LICENSE_S-aiueo32
rename to GanInverter/licenses/LICENSE_S-aiueo32
diff --git a/model/GanInverter/licenses/LICENSE_TreB1eN b/GanInverter/licenses/LICENSE_TreB1eN
similarity index 100%
rename from model/GanInverter/licenses/LICENSE_TreB1eN
rename to GanInverter/licenses/LICENSE_TreB1eN
diff --git a/model/GanInverter/licenses/LICENSE_lessw2020 b/GanInverter/licenses/LICENSE_lessw2020
similarity index 100%
rename from model/GanInverter/licenses/LICENSE_lessw2020
rename to GanInverter/licenses/LICENSE_lessw2020
diff --git a/model/GanInverter/licenses/LICENSE_rosinality b/GanInverter/licenses/LICENSE_rosinality
similarity index 100%
rename from model/GanInverter/licenses/LICENSE_rosinality
rename to GanInverter/licenses/LICENSE_rosinality
diff --git a/model/GanInverter/models/__init__.py b/GanInverter/models/__init__.py
similarity index 100%
rename from model/GanInverter/models/__init__.py
rename to GanInverter/models/__init__.py
diff --git a/model/GanInverter/models/bisenet/__init__.py b/GanInverter/models/bisenet/__init__.py
similarity index 100%
rename from model/GanInverter/models/bisenet/__init__.py
rename to GanInverter/models/bisenet/__init__.py
diff --git a/model/GanInverter/models/bisenet/model.py b/GanInverter/models/bisenet/model.py
similarity index 100%
rename from model/GanInverter/models/bisenet/model.py
rename to GanInverter/models/bisenet/model.py
diff --git a/model/GanInverter/models/bisenet/resnet.py b/GanInverter/models/bisenet/resnet.py
similarity index 100%
rename from model/GanInverter/models/bisenet/resnet.py
rename to GanInverter/models/bisenet/resnet.py
diff --git a/model/GanInverter/models/discriminator.py b/GanInverter/models/discriminator.py
similarity index 100%
rename from model/GanInverter/models/discriminator.py
rename to GanInverter/models/discriminator.py
diff --git a/model/GanInverter/models/encoder.py b/GanInverter/models/encoder.py
similarity index 100%
rename from model/GanInverter/models/encoder.py
rename to GanInverter/models/encoder.py
diff --git a/model/GanInverter/models/encoders/.DS_Store b/GanInverter/models/encoders/.DS_Store
similarity index 100%
rename from model/GanInverter/models/encoders/.DS_Store
rename to GanInverter/models/encoders/.DS_Store
diff --git a/model/GanInverter/models/encoders/__init__.py b/GanInverter/models/encoders/__init__.py
similarity index 100%
rename from model/GanInverter/models/encoders/__init__.py
rename to GanInverter/models/encoders/__init__.py
diff --git a/model/GanInverter/models/encoders/helpers.py b/GanInverter/models/encoders/helpers.py
similarity index 100%
rename from model/GanInverter/models/encoders/helpers.py
rename to GanInverter/models/encoders/helpers.py
diff --git a/model/GanInverter/models/encoders/model_irse.py b/GanInverter/models/encoders/model_irse.py
similarity index 100%
rename from model/GanInverter/models/encoders/model_irse.py
rename to GanInverter/models/encoders/model_irse.py
diff --git a/model/GanInverter/models/encoders/psp_encoders.py b/GanInverter/models/encoders/psp_encoders.py
similarity index 100%
rename from model/GanInverter/models/encoders/psp_encoders.py
rename to GanInverter/models/encoders/psp_encoders.py
diff --git a/model/GanInverter/models/invertibility/__init__.py b/GanInverter/models/invertibility/__init__.py
similarity index 100%
rename from model/GanInverter/models/invertibility/__init__.py
rename to GanInverter/models/invertibility/__init__.py
diff --git a/model/GanInverter/models/invertibility/aspp.py b/GanInverter/models/invertibility/aspp.py
similarity index 100%
rename from model/GanInverter/models/invertibility/aspp.py
rename to GanInverter/models/invertibility/aspp.py
diff --git a/model/GanInverter/models/invertibility/backbone/__init__.py b/GanInverter/models/invertibility/backbone/__init__.py
similarity index 100%
rename from model/GanInverter/models/invertibility/backbone/__init__.py
rename to GanInverter/models/invertibility/backbone/__init__.py
diff --git a/model/GanInverter/models/invertibility/backbone/drn.py b/GanInverter/models/invertibility/backbone/drn.py
similarity index 100%
rename from model/GanInverter/models/invertibility/backbone/drn.py
rename to GanInverter/models/invertibility/backbone/drn.py
diff --git a/model/GanInverter/models/invertibility/backbone/mobilenet.py b/GanInverter/models/invertibility/backbone/mobilenet.py
similarity index 100%
rename from model/GanInverter/models/invertibility/backbone/mobilenet.py
rename to GanInverter/models/invertibility/backbone/mobilenet.py
diff --git a/model/GanInverter/models/invertibility/backbone/resnet.py b/GanInverter/models/invertibility/backbone/resnet.py
similarity index 100%
rename from model/GanInverter/models/invertibility/backbone/resnet.py
rename to GanInverter/models/invertibility/backbone/resnet.py
diff --git a/model/GanInverter/models/invertibility/backbone/xception.py b/GanInverter/models/invertibility/backbone/xception.py
similarity index 100%
rename from model/GanInverter/models/invertibility/backbone/xception.py
rename to GanInverter/models/invertibility/backbone/xception.py
diff --git a/model/GanInverter/models/invertibility/decoder.py b/GanInverter/models/invertibility/decoder.py
similarity index 100%
rename from model/GanInverter/models/invertibility/decoder.py
rename to GanInverter/models/invertibility/decoder.py
diff --git a/model/GanInverter/models/invertibility/deeplab.py b/GanInverter/models/invertibility/deeplab.py
similarity index 100%
rename from model/GanInverter/models/invertibility/deeplab.py
rename to GanInverter/models/invertibility/deeplab.py
diff --git a/model/GanInverter/models/invertibility/sync_batchnorm/__init__.py b/GanInverter/models/invertibility/sync_batchnorm/__init__.py
similarity index 100%
rename from model/GanInverter/models/invertibility/sync_batchnorm/__init__.py
rename to GanInverter/models/invertibility/sync_batchnorm/__init__.py
diff --git a/model/GanInverter/models/invertibility/sync_batchnorm/batchnorm.py b/GanInverter/models/invertibility/sync_batchnorm/batchnorm.py
similarity index 100%
rename from model/GanInverter/models/invertibility/sync_batchnorm/batchnorm.py
rename to GanInverter/models/invertibility/sync_batchnorm/batchnorm.py
diff --git a/model/GanInverter/models/invertibility/sync_batchnorm/comm.py b/GanInverter/models/invertibility/sync_batchnorm/comm.py
similarity index 100%
rename from model/GanInverter/models/invertibility/sync_batchnorm/comm.py
rename to GanInverter/models/invertibility/sync_batchnorm/comm.py
diff --git a/model/GanInverter/models/invertibility/sync_batchnorm/replicate.py b/GanInverter/models/invertibility/sync_batchnorm/replicate.py
similarity index 100%
rename from model/GanInverter/models/invertibility/sync_batchnorm/replicate.py
rename to GanInverter/models/invertibility/sync_batchnorm/replicate.py
diff --git a/model/GanInverter/models/invertibility/sync_batchnorm/unittest.py b/GanInverter/models/invertibility/sync_batchnorm/unittest.py
similarity index 100%
rename from model/GanInverter/models/invertibility/sync_batchnorm/unittest.py
rename to GanInverter/models/invertibility/sync_batchnorm/unittest.py
diff --git a/model/GanInverter/models/latent_codes_pool.py b/GanInverter/models/latent_codes_pool.py
similarity index 100%
rename from model/GanInverter/models/latent_codes_pool.py
rename to GanInverter/models/latent_codes_pool.py
diff --git a/model/GanInverter/models/segmenter.py b/GanInverter/models/segmenter.py
similarity index 100%
rename from model/GanInverter/models/segmenter.py
rename to GanInverter/models/segmenter.py
diff --git a/model/GanInverter/models/stylegan2/__init__.py b/GanInverter/models/stylegan2/__init__.py
similarity index 100%
rename from model/GanInverter/models/stylegan2/__init__.py
rename to GanInverter/models/stylegan2/__init__.py
diff --git a/model/GanInverter/models/stylegan2/model.py b/GanInverter/models/stylegan2/model.py
similarity index 100%
rename from model/GanInverter/models/stylegan2/model.py
rename to GanInverter/models/stylegan2/model.py
diff --git a/model/GanInverter/models/stylegan2/op/__init__.py b/GanInverter/models/stylegan2/op/__init__.py
similarity index 100%
rename from model/GanInverter/models/stylegan2/op/__init__.py
rename to GanInverter/models/stylegan2/op/__init__.py
diff --git a/model/GanInverter/models/stylegan2/op/fused_act.py b/GanInverter/models/stylegan2/op/fused_act.py
similarity index 100%
rename from model/GanInverter/models/stylegan2/op/fused_act.py
rename to GanInverter/models/stylegan2/op/fused_act.py
diff --git a/model/GanInverter/models/stylegan2/op/fused_bias_act.cpp b/GanInverter/models/stylegan2/op/fused_bias_act.cpp
similarity index 100%
rename from model/GanInverter/models/stylegan2/op/fused_bias_act.cpp
rename to GanInverter/models/stylegan2/op/fused_bias_act.cpp
diff --git a/model/GanInverter/models/stylegan2/op/fused_bias_act_kernel.cu b/GanInverter/models/stylegan2/op/fused_bias_act_kernel.cu
similarity index 100%
rename from model/GanInverter/models/stylegan2/op/fused_bias_act_kernel.cu
rename to GanInverter/models/stylegan2/op/fused_bias_act_kernel.cu
diff --git a/model/GanInverter/models/stylegan2/op/upfirdn2d.cpp b/GanInverter/models/stylegan2/op/upfirdn2d.cpp
similarity index 100%
rename from model/GanInverter/models/stylegan2/op/upfirdn2d.cpp
rename to GanInverter/models/stylegan2/op/upfirdn2d.cpp
diff --git a/model/GanInverter/models/stylegan2/op/upfirdn2d.py b/GanInverter/models/stylegan2/op/upfirdn2d.py
similarity index 100%
rename from model/GanInverter/models/stylegan2/op/upfirdn2d.py
rename to GanInverter/models/stylegan2/op/upfirdn2d.py
diff --git a/model/GanInverter/models/stylegan2/op/upfirdn2d_kernel.cu b/GanInverter/models/stylegan2/op/upfirdn2d_kernel.cu
similarity index 100%
rename from model/GanInverter/models/stylegan2/op/upfirdn2d_kernel.cu
rename to GanInverter/models/stylegan2/op/upfirdn2d_kernel.cu
diff --git a/model/GanInverter/options/__init__.py b/GanInverter/options/__init__.py
similarity index 100%
rename from model/GanInverter/options/__init__.py
rename to GanInverter/options/__init__.py
diff --git a/model/GanInverter/options/base_options.py b/GanInverter/options/base_options.py
similarity index 100%
rename from model/GanInverter/options/base_options.py
rename to GanInverter/options/base_options.py
diff --git a/model/GanInverter/options/test_options.py b/GanInverter/options/test_options.py
similarity index 100%
rename from model/GanInverter/options/test_options.py
rename to GanInverter/options/test_options.py
diff --git a/model/GanInverter/options/train_options.py b/GanInverter/options/train_options.py
similarity index 100%
rename from model/GanInverter/options/train_options.py
rename to GanInverter/options/train_options.py
diff --git a/model/GanInverter/requirements.txt b/GanInverter/requirements.txt
similarity index 100%
rename from model/GanInverter/requirements.txt
rename to GanInverter/requirements.txt
diff --git a/model/GanInverter/scripts/edit.py b/GanInverter/scripts/edit.py
similarity index 100%
rename from model/GanInverter/scripts/edit.py
rename to GanInverter/scripts/edit.py
diff --git a/model/GanInverter/scripts/infer.py b/GanInverter/scripts/infer.py
similarity index 100%
rename from model/GanInverter/scripts/infer.py
rename to GanInverter/scripts/infer.py
diff --git a/model/GanInverter/scripts/test.py b/GanInverter/scripts/test.py
similarity index 100%
rename from model/GanInverter/scripts/test.py
rename to GanInverter/scripts/test.py
diff --git a/model/GanInverter/scripts/train.py b/GanInverter/scripts/train.py
similarity index 100%
rename from model/GanInverter/scripts/train.py
rename to GanInverter/scripts/train.py
diff --git a/model/GanInverter/training/__init__.py b/GanInverter/training/__init__.py
similarity index 100%
rename from model/GanInverter/training/__init__.py
rename to GanInverter/training/__init__.py
diff --git a/model/GanInverter/training/encoder_trainer.py b/GanInverter/training/encoder_trainer.py
similarity index 100%
rename from model/GanInverter/training/encoder_trainer.py
rename to GanInverter/training/encoder_trainer.py
diff --git a/model/GanInverter/utils/__init__.py b/GanInverter/utils/__init__.py
similarity index 100%
rename from model/GanInverter/utils/__init__.py
rename to GanInverter/utils/__init__.py
diff --git a/model/GanInverter/utils/common.py b/GanInverter/utils/common.py
similarity index 100%
rename from model/GanInverter/utils/common.py
rename to GanInverter/utils/common.py
diff --git a/model/GanInverter/utils/data_utils.py b/GanInverter/utils/data_utils.py
similarity index 100%
rename from model/GanInverter/utils/data_utils.py
rename to GanInverter/utils/data_utils.py
diff --git a/model/GanInverter/utils/dist.py b/GanInverter/utils/dist.py
similarity index 100%
rename from model/GanInverter/utils/dist.py
rename to GanInverter/utils/dist.py
diff --git a/model/GanInverter/utils/facer/LICENSE b/GanInverter/utils/facer/LICENSE
similarity index 100%
rename from model/GanInverter/utils/facer/LICENSE
rename to GanInverter/utils/facer/LICENSE
diff --git a/model/GanInverter/utils/facer/README.md b/GanInverter/utils/facer/README.md
similarity index 100%
rename from model/GanInverter/utils/facer/README.md
rename to GanInverter/utils/facer/README.md
diff --git a/model/GanInverter/utils/facer/facer/__init__.py b/GanInverter/utils/facer/facer/__init__.py
similarity index 100%
rename from model/GanInverter/utils/facer/facer/__init__.py
rename to GanInverter/utils/facer/facer/__init__.py
diff --git a/model/GanInverter/utils/facer/facer/draw.py b/GanInverter/utils/facer/facer/draw.py
similarity index 100%
rename from model/GanInverter/utils/facer/facer/draw.py
rename to GanInverter/utils/facer/facer/draw.py
diff --git a/model/GanInverter/utils/facer/facer/face_detection/__init__.py b/GanInverter/utils/facer/facer/face_detection/__init__.py
similarity index 100%
rename from model/GanInverter/utils/facer/facer/face_detection/__init__.py
rename to GanInverter/utils/facer/facer/face_detection/__init__.py
diff --git a/model/GanInverter/utils/facer/facer/face_detection/base.py b/GanInverter/utils/facer/facer/face_detection/base.py
similarity index 100%
rename from model/GanInverter/utils/facer/facer/face_detection/base.py
rename to GanInverter/utils/facer/facer/face_detection/base.py
diff --git a/model/GanInverter/utils/facer/facer/face_detection/retinaface.py b/GanInverter/utils/facer/facer/face_detection/retinaface.py
similarity index 100%
rename from model/GanInverter/utils/facer/facer/face_detection/retinaface.py
rename to GanInverter/utils/facer/facer/face_detection/retinaface.py
diff --git a/model/GanInverter/utils/facer/facer/face_landmark/__init__.py b/GanInverter/utils/facer/facer/face_landmark/__init__.py
similarity index 100%
rename from model/GanInverter/utils/facer/facer/face_landmark/__init__.py
rename to GanInverter/utils/facer/facer/face_landmark/__init__.py
diff --git a/model/GanInverter/utils/facer/facer/face_landmark/base.py b/GanInverter/utils/facer/facer/face_landmark/base.py
similarity index 100%
rename from model/GanInverter/utils/facer/facer/face_landmark/base.py
rename to GanInverter/utils/facer/facer/face_landmark/base.py
diff --git a/model/GanInverter/utils/facer/facer/face_parsing/__init__.py b/GanInverter/utils/facer/facer/face_parsing/__init__.py
similarity index 100%
rename from model/GanInverter/utils/facer/facer/face_parsing/__init__.py
rename to GanInverter/utils/facer/facer/face_parsing/__init__.py
diff --git a/model/GanInverter/utils/facer/facer/face_parsing/base.py b/GanInverter/utils/facer/facer/face_parsing/base.py
similarity index 100%
rename from model/GanInverter/utils/facer/facer/face_parsing/base.py
rename to GanInverter/utils/facer/facer/face_parsing/base.py
diff --git a/model/GanInverter/utils/facer/facer/face_parsing/farl.py b/GanInverter/utils/facer/facer/face_parsing/farl.py
similarity index 100%
rename from model/GanInverter/utils/facer/facer/face_parsing/farl.py
rename to GanInverter/utils/facer/facer/face_parsing/farl.py
diff --git a/model/GanInverter/utils/facer/facer/io.py b/GanInverter/utils/facer/facer/io.py
similarity index 100%
rename from model/GanInverter/utils/facer/facer/io.py
rename to GanInverter/utils/facer/facer/io.py
diff --git a/model/GanInverter/utils/facer/facer/show.py b/GanInverter/utils/facer/facer/show.py
similarity index 100%
rename from model/GanInverter/utils/facer/facer/show.py
rename to GanInverter/utils/facer/facer/show.py
diff --git a/model/GanInverter/utils/facer/facer/transform.py b/GanInverter/utils/facer/facer/transform.py
similarity index 100%
rename from model/GanInverter/utils/facer/facer/transform.py
rename to GanInverter/utils/facer/facer/transform.py
diff --git a/model/GanInverter/utils/facer/facer/util.py b/GanInverter/utils/facer/facer/util.py
similarity index 100%
rename from model/GanInverter/utils/facer/facer/util.py
rename to GanInverter/utils/facer/facer/util.py
diff --git a/model/GanInverter/utils/facer/facer/version.py b/GanInverter/utils/facer/facer/version.py
similarity index 100%
rename from model/GanInverter/utils/facer/facer/version.py
rename to GanInverter/utils/facer/facer/version.py
diff --git a/model/GanInverter/utils/facer/requirements.txt b/GanInverter/utils/facer/requirements.txt
similarity index 100%
rename from model/GanInverter/utils/facer/requirements.txt
rename to GanInverter/utils/facer/requirements.txt
diff --git a/model/GanInverter/utils/facer/samples/data/fire.webp b/GanInverter/utils/facer/samples/data/fire.webp
similarity index 100%
rename from model/GanInverter/utils/facer/samples/data/fire.webp
rename to GanInverter/utils/facer/samples/data/fire.webp
diff --git a/model/GanInverter/utils/facer/samples/data/girl.jpg b/GanInverter/utils/facer/samples/data/girl.jpg
similarity index 100%
rename from model/GanInverter/utils/facer/samples/data/girl.jpg
rename to GanInverter/utils/facer/samples/data/girl.jpg
diff --git a/model/GanInverter/utils/facer/samples/data/sideface.jpg b/GanInverter/utils/facer/samples/data/sideface.jpg
similarity index 100%
rename from model/GanInverter/utils/facer/samples/data/sideface.jpg
rename to GanInverter/utils/facer/samples/data/sideface.jpg
diff --git a/model/GanInverter/utils/facer/samples/data/twogirls.jpg b/GanInverter/utils/facer/samples/data/twogirls.jpg
similarity index 100%
rename from model/GanInverter/utils/facer/samples/data/twogirls.jpg
rename to GanInverter/utils/facer/samples/data/twogirls.jpg
diff --git a/model/GanInverter/utils/facer/samples/data/weirdface.jpg b/GanInverter/utils/facer/samples/data/weirdface.jpg
similarity index 100%
rename from model/GanInverter/utils/facer/samples/data/weirdface.jpg
rename to GanInverter/utils/facer/samples/data/weirdface.jpg
diff --git a/model/GanInverter/utils/facer/samples/data/weirdface2.jpg b/GanInverter/utils/facer/samples/data/weirdface2.jpg
similarity index 100%
rename from model/GanInverter/utils/facer/samples/data/weirdface2.jpg
rename to GanInverter/utils/facer/samples/data/weirdface2.jpg
diff --git a/model/GanInverter/utils/facer/samples/data/weirdface3.jpg b/GanInverter/utils/facer/samples/data/weirdface3.jpg
similarity index 100%
rename from model/GanInverter/utils/facer/samples/data/weirdface3.jpg
rename to GanInverter/utils/facer/samples/data/weirdface3.jpg
diff --git a/model/GanInverter/utils/facer/samples/face_detect.ipynb b/GanInverter/utils/facer/samples/face_detect.ipynb
similarity index 100%
rename from model/GanInverter/utils/facer/samples/face_detect.ipynb
rename to GanInverter/utils/facer/samples/face_detect.ipynb
diff --git a/model/GanInverter/utils/facer/samples/face_parsing.ipynb b/GanInverter/utils/facer/samples/face_parsing.ipynb
similarity index 100%
rename from model/GanInverter/utils/facer/samples/face_parsing.ipynb
rename to GanInverter/utils/facer/samples/face_parsing.ipynb
diff --git a/model/GanInverter/utils/facer/samples/transform.ipynb b/GanInverter/utils/facer/samples/transform.ipynb
similarity index 100%
rename from model/GanInverter/utils/facer/samples/transform.ipynb
rename to GanInverter/utils/facer/samples/transform.ipynb
diff --git a/model/GanInverter/utils/facer/scripts/build.sh b/GanInverter/utils/facer/scripts/build.sh
similarity index 100%
rename from model/GanInverter/utils/facer/scripts/build.sh
rename to GanInverter/utils/facer/scripts/build.sh
diff --git a/model/GanInverter/utils/facer/scripts/publish.sh b/GanInverter/utils/facer/scripts/publish.sh
similarity index 100%
rename from model/GanInverter/utils/facer/scripts/publish.sh
rename to GanInverter/utils/facer/scripts/publish.sh
diff --git a/model/GanInverter/utils/facer/setup.py b/GanInverter/utils/facer/setup.py
similarity index 100%
rename from model/GanInverter/utils/facer/setup.py
rename to GanInverter/utils/facer/setup.py
diff --git a/model/GanInverter/utils/ranger.py b/GanInverter/utils/ranger.py
similarity index 100%
rename from model/GanInverter/utils/ranger.py
rename to GanInverter/utils/ranger.py
diff --git a/model/GanInverter/utils/train_utils.py b/GanInverter/utils/train_utils.py
similarity index 100%
rename from model/GanInverter/utils/train_utils.py
rename to GanInverter/utils/train_utils.py
diff --git a/model/GanInverter/utils/wandb_utils.py b/GanInverter/utils/wandb_utils.py
similarity index 100%
rename from model/GanInverter/utils/wandb_utils.py
rename to GanInverter/utils/wandb_utils.py
diff --git a/model/GanInverter/editing/interfacegan_directions/age.pt b/model/GanInverter/editing/interfacegan_directions/age.pt
deleted file mode 100644
index 73b4e6c..0000000
Binary files a/model/GanInverter/editing/interfacegan_directions/age.pt and /dev/null differ
diff --git a/model/GanInverter/editing/interfacegan_directions/pose.pt b/model/GanInverter/editing/interfacegan_directions/pose.pt
deleted file mode 100644
index aa828a4..0000000
Binary files a/model/GanInverter/editing/interfacegan_directions/pose.pt and /dev/null differ
diff --git a/model/GanInverter/editing/interfacegan_directions/smile.pt b/model/GanInverter/editing/interfacegan_directions/smile.pt
deleted file mode 100644
index 3c44456..0000000
Binary files a/model/GanInverter/editing/interfacegan_directions/smile.pt and /dev/null differ