Update app.py

app.py

@@ -204,7 +204,10 @@ def merge_incompatible_lora(full_path_lora, lora_scale):
                 del weights_sd
                 del lora_model
 @spaces.GPU
-def generate_image(prompt, negative, face_emb, face_image, face_kps, image_strength, guidance_scale, face_strength, depth_control_scale, repo_name, loaded_state_dict, lora_scale, sdxl_loras, selected_state_index):
+def generate_image(prompt, negative, face_emb, face_image, face_kps, image_strength, guidance_scale, face_strength, depth_control_scale, repo_name, loaded_state_dict, lora_scale, sdxl_loras, selected_state_index, st):
+    et = time.time()
+    elapsed_time = et - st
+    print('Getting into the decorated function took: ', elapsed_time, 'seconds')
     global last_fused, last_lora
     print("Last LoRA: ", last_lora)
     print("Current LoRA: ", repo_name)
@@ -259,6 +262,7 @@ def generate_image(prompt, negative, face_emb, face_image, face_kps, image_stren
     elapsed_time = et - st
     print('Prompt processing took: ', elapsed_time, 'seconds')
     print("Processing image...")
+    st = time.time()
     image = pipe(
         prompt_embeds=conditioning,
         pooled_prompt_embeds=pooled,
@@ -274,6 +278,9 @@ def generate_image(prompt, negative, face_emb, face_image, face_kps, image_stren
         guidance_scale = guidance_scale,
         controlnet_conditioning_scale=[face_strength, depth_control_scale],
     ).images[0]
+    et = time.time()
+    elapsed_time = et - st
+    print('Image processing took: ', elapsed_time, 'seconds')
     last_lora = repo_name
     return image
     
@@ -320,8 +327,9 @@ def run_lora(face_image, prompt, negative, lora_scale, selected_state, face_stre
     et = time.time()
     elapsed_time = et - st
     print('Small content processing took: ', elapsed_time, 'seconds')
-    
-    image = generate_image(prompt, negative, face_emb, face_image, face_kps, image_strength, guidance_scale, face_strength, depth_control_scale, repo_name, loaded_state_dict, lora_scale, sdxl_loras, selected_state_index)
+
+    st = time.time()
+    image = generate_image(prompt, negative, face_emb, face_image, face_kps, image_strength, guidance_scale, face_strength, depth_control_scale, repo_name, loaded_state_dict, lora_scale, sdxl_loras, selected_state_index, st)
     return image, gr.update(visible=True)
 
 def shuffle_gallery(sdxl_loras):