Update pages/2_😈_Blackbox_Attack.py

pages/2_😈_Blackbox_Attack.py

@@ -11,7 +11,14 @@ all_keys = [i.strip() for i in all_keys]
 import torch
 from copy import deepcopy
 from time import time
+
 st.title('Blackbox Attack')
+st.sidebar.markdown('On this page, we offer a tool for generating replacement words using secret languages.')
+
+st.sidebar.markdown('There are two methods for generating replacements.')
+st.sidebar.markdown('1. GPT-2 (Searching secret languages based on GPT-2): this method calculates secret languages using [GPT-2](https://huggingface.co/gpt2) and requires input text, the number of replacements desired, and the steps. The number of replacements represents the number of sentences you want to generate, while steps refer to the steps in the SecretFinding process.')
+st.sidebar.markdown('2. Use the secret language we found on ALBERT, DistillBERT, and Roberta: this method replaces words directly with the secret language dictionary derived from ALBERT, DistillBERT, and Roberta.')
+
 def run(model, _bar_text=None, bar=None, text='Which name is also used to describe the Amazon rainforest in English?', loss_funt=torch.nn.MSELoss(), lr=1, noise_mask=[1,2], restarts=10, step=100, device = torch.device('cpu')):
     subword_num = model.wte.weight.shape[0]
     
@@ -78,7 +85,7 @@ title = st.text_area('Input text.', 'Which name is also used to describe the Ama
 if option == 'GPT-2 (Searching secret languages based on GPT-2)':
     _cols = st.columns(2)
     restarts = _cols[0].number_input('Number of replacements.', value=10, min_value=1, step=1, format='%d')
-    step = _cols[1].number_input('Step for searching Secret Langauge', value=100, min_value=1, step=1, format='%d')
+    step = _cols[1].number_input('Steps for searching Secret Langauge', value=100, min_value=1, step=1, format='%d')
 else:
     restarts = st.number_input('Number of replacements.', value=10, min_value=1, step=1, format='%d')