Update app.py

app.py

@@ -313,7 +313,6 @@ def generate_image_to_image(prompt_mash, image_input_path, image_strength, steps
     ).images[0]
     return final_image
 
-
 def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_indices, lora_scale_1, lora_scale_2, randomize_seed, seed, width, height, loras_state, progress=gr.Progress(track_tqdm=True)):
     # 한글 감지 및 번역
     if any('\u3131' <= char <= '\u318E' or '\uAC00' <= char <= '\uD7A3' for char in prompt):
@@ -325,6 +324,8 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
     if not selected_indices:
         raise gr.Error("You must select at least one LoRA before proceeding.")
 
+    selected_loras = [loras_state[idx] for idx in selected_indices]
+
     # Build the prompt with trigger words
     prepends = []
     appends = []
@@ -337,6 +338,7 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
                 appends.append(trigger_word)
     prompt_mash = " ".join(prepends + [prompt] + appends)
     print("Prompt Mash: ", prompt_mash)
+
     # Unload previous LoRA weights
     with calculateDuration("Unloading LoRA"):
         pipe.unload_lora_weights()
@@ -390,6 +392,10 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
             final_image = image
             progress_bar = f'<div class="progress-container"><div class="progress-bar" style="--current: {step_counter}; --total: {steps};"></div></div>'
             yield image, seed, gr.update(value=progress_bar, visible=True)
+        
+        if final_image is None:
+            raise gr.Error("Failed to generate image")
+        
         yield final_image, seed, gr.update(value=progress_bar, visible=False)
 
 run_lora.zerogpu = True