fix import

app.py

@@ -19,7 +19,7 @@ from os.path import join as opj
 token = os.getenv("ACCESS_TOKEN")
 os.system(f"python -m pip install git+https://{token}@github.com/logn-2024/StableGarment.git")
 
-from stablegarment.models import AppearanceEncoderModel,ControlNetModel
+from stablegarment.models import GarmentEncoderModel,ControlNetModel
 from stablegarment.piplines import StableGarmentPipeline,StableGarmentControlNetPipeline
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -32,7 +32,7 @@ vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse").to(dtype=torch_
 scheduler = UniPCMultistepScheduler.from_pretrained("runwayml/stable-diffusion-v1-5",subfolder="scheduler")
 
 pretrained_garment_encoder_path = "loooooong/StableGarment_text2img"
-garment_encoder = AppearanceEncoderModel.from_pretrained(pretrained_garment_encoder_path,torch_dtype=torch_dtype,subfolder="garment_encoder")
+garment_encoder = GarmentEncoderModel.from_pretrained(pretrained_garment_encoder_path,torch_dtype=torch_dtype,subfolder="garment_encoder")
 garment_encoder = garment_encoder.to(device=device,dtype=torch_dtype)
 
 pipeline_t2i = StableGarmentPipeline.from_pretrained(base_model_path, vae=vae, torch_dtype=torch_dtype,).to(device=device) #  variant="fp16"