Update app.py

app.py

@@ -207,7 +207,12 @@ def merge_incompatible_lora(full_path_lora, lora_scale):
                 del lora_model
                 gc.collect()
 @spaces.GPU
-def generate_image(conditioning, pooled, negative_conditioning, negative_pooled, face_emb, face_image, image_strength, images, guidance_scale, face_strength, depth_control_scale):
+def generate_image(prompt, negative, face_emb, face_image, image_strength, images, guidance_scale, face_strength, depth_control_scale):
+    conditioning, pooled = compel(prompt)
+    if(negative):
+        negative_conditioning, negative_pooled = compel(negative)
+    else:
+        negative_conditioning, negative_pooled = None, None
     image = pipe(
         prompt_embeds=conditioning,
         pooled_prompt_embeds=pooled,
@@ -300,13 +305,7 @@ def run_lora(face_image, prompt, negative, lora_scale, selected_state, face_stre
                 pipe.load_textual_inversion(state_dict_embedding["text_encoders_0"], token=["<s0>", "<s1>"], text_encoder=pipe.text_encoder, tokenizer=pipe.tokenizer)
                 pipe.load_textual_inversion(state_dict_embedding["text_encoders_1"], token=["<s0>", "<s1>"], text_encoder=pipe.text_encoder_2, tokenizer=pipe.tokenizer_2)
 
-    conditioning, pooled = compel(prompt)
-    if(negative):
-        negative_conditioning, negative_pooled = compel(negative)
-    else:
-        negative_conditioning, negative_pooled = None, None
-
-    image = generate_image(conditioning, pooled, negative_conditioning, negative_pooled, face_emb, face_image, image_strength, images, guidance_scale, face_strength, depth_control_scale)
+    image = generate_image(prompt, negative, face_emb, face_image, image_strength, images, guidance_scale, face_strength, depth_control_scale)
     last_lora = repo_name
     gc.collect()
     return image, gr.update(visible=True)