no head selected fix

app.py

@@ -59,8 +59,7 @@ def run_unicorn_ablation(selected_layer, selected_heads):
     # layer = selected_layer.split('blocks.')[1].split('.attentions')[0]   
     if selected_heads is None or len(selected_heads) ==0:
         selected_heads = []
-        UNICORN_NO_HORN_PATH = "unicorn_with_horn.png"
-        return UNICORN_WITH_HORN_PATH, UNICORN_NO_HORN_PATH    
+        return UNICORN_WITH_HORN_PATH    
     if len(selected_heads) > 3:
         selected_heads = selected_heads[:3]
     selected_heads.sort()    
@@ -75,7 +74,7 @@ def run_unicorn_ablation(selected_layer, selected_heads):
     UNICORN_NO_HORN_PATH = path
     print(f"Unicorn no horn path was {UNICORN_NO_HORN_PATH}")
     # In a real experiment you'd use `selected_heads` to ablate SDXL heads.
-    return UNICORN_WITH_HORN_PATH, UNICORN_NO_HORN_PATH
+    return UNICORN_NO_HORN_PATH
 
 
 with gr.Blocks() as demo: