fix bugs

app.py

@@ -92,11 +92,11 @@ def retrieve_transferability_information():
 
     data = retrieve_transfer(source_model_name, target_model_name, attack_name, shot)
     for d in data:
-        st.expander(f"Dataset: {d['dataset']} Prompt Type: {d['type']}-oriented")
-        st.write(f"Origin prompt: {d['origin_prompt']}")
-        st.write(f"Attack prompt: {d['atk_prompt']}")
-        st.write(f"Source model: origin acc: {d['origin_acc']}, attack acc: {d['atk_acc']}")
-        st.write(f"Target model: origin acc: {d['transfer_ori_acc']}, attack acc: {d['transfer_atk_acc']}")
+        with st.expander(f"Dataset: {d['dataset']} Prompt Type: {d['type']}-oriented"):
+            st.write(f"Origin prompt: {d['origin_prompt']}")
+            st.write(f"Attack prompt: {d['atk_prompt']}")
+            st.write(f"Source model: origin acc: {d['origin_acc']}, attack acc: {d['atk_acc']}")
+            st.write(f"Target model: origin acc: {d['transfer_ori_acc']}, attack acc: {d['transfer_atk_acc']}")
 
 if __name__ == "__main__":
     main()