fix bugs

app.py

@@ -92,12 +92,11 @@ def retrieve_transferability_information():
 
     data = retrieve_transfer(source_model_name, target_model_name, attack_name, shot)
     for d in data:
-        st.write(f"Dataset: {d['dataset']}")
-        st.write(f"Prompt Type: {d['type']}-oriented")
+        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_origin_acc']}, attack acc: {d['transfer_atk_acc']}")
+        st.write(f"Target model: origin acc: {d['transfer_ori_acc']}, attack acc: {d['transfer_atk_acc']}")
 
 if __name__ == "__main__":
     main()