Update app.py

app.py

@@ -60,10 +60,10 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, wid
         joint_attention_kwargs={"scale": lora_scale},
     ).images[0]
 
-    # Unload LoRA weights
+    yield image
+    
     pipe.unload_lora_weights()
-
-    return image
+    
 
 '''
 #gen_column{height: 100%}