Update app.py

app.py

@@ -21,10 +21,27 @@ pipe_normal.load_textual_inversion(state_dict["clip_l"], token="unaestheticXLv31
 
 pipe_normal.to("cuda")
 
+compel = Compel(tokenizer=[pipe_normal.tokenizer, pipe_normal.tokenizer_2] , 
+                text_encoder=[pipe_normal.text_encoder, pipe_normal.text_encoder_2], 
+                returned_embeddings_type=ReturnedEmbeddingsType.PENULTIMATE_HIDDEN_STATES_NON_NORMALIZED, 
+                requires_pooled=[False, True])
 
 @spaces.GPU
 def run_normal(prompt, negative_prompt="", guidance_scale=7.5, progress=gr.Progress(track_tqdm=True)):
-    return pipe_normal(prompt, negative_prompt="unaestheticXLv31"+negative_prompt, guidance_scale=guidance_scale, num_inference_steps=20).images[0]
+    conditioning, pooled = compel([prompt, "unaestheticXLv31--, "+neg_prompt])
+    
+    result = pipe(
+        prompt_embeds=conditioning[0:1],
+        pooled_prompt_embeds=pooled[0:1], 
+        negative_prompt_embeds=conditioning[1:2], 
+        negative_pooled_prompt_embeds=pooled[1:2],
+        num_inference_steps = int(steps),
+        guidance_scale = guidance,
+        width = width,
+        height = height,
+        generator = generator)
+    
+    return result.images[0]
 
 css = '''
 .gradio-container{