Update app.py

app.py

@@ -21,6 +21,10 @@ state_dict = load_file(negative_ti_file)
 pipe_normal.load_textual_inversion(state_dict["clip_g"], token="unaestheticXLv31", text_encoder=pipe_normal.text_encoder_2, tokenizer=pipe_normal.tokenizer_2)
 pipe_normal.load_textual_inversion(state_dict["clip_l"], token="unaestheticXLv31", text_encoder=pipe_normal.text_encoder, tokenizer=pipe_normal.tokenizer)
 
+state_dict = load_file("unaestheticXL_Alb2.safetensors")
+pipe_normal.load_textual_inversion(state_dict["clip_g"], token="unaestheticXL_Alb2", text_encoder=pipe_normal.text_encoder_2, tokenizer=pipe_normal.tokenizer_2)
+pipe_normal.load_textual_inversion(state_dict["clip_l"], token="unaestheticXL_Alb2", text_encoder=pipe_normal.text_encoder, tokenizer=pipe_normal.tokenizer)
+
 pipe_normal.to("cuda")
 
 pipe_normal.enable_freeu(s1=1.2, s2=0.7, b1=1.1, b2=1.3)
@@ -40,7 +44,7 @@ def run_normal(prompt, negative_prompt="", guidance_scale=7.5, progress=gr.Progr
         conditioning, pooled = compel("1girl, (upper body)++, brown bob short hair, brown eyes, looking at viewer, cherry blossom")
     else:
         conditioning, pooled = compel(prompt)
-    negative_conditioning, negatice_pooled = compel("(unaestheticXLv31)++++++, (bad hands), bad anatomy, low quality, 3d, photo, realism, text, sign, "+negative_prompt)
+    negative_conditioning, negatice_pooled = compel("(unaestheticXLv31)++, (unaestheticXL_Alb2)++, bad hands, bad anatomy, low quality, 3d, photo, realism, text, sign, "+negative_prompt)
     result = pipe_normal(
         prompt_embeds=conditioning,
         pooled_prompt_embeds=pooled,