anonymousauthors commited on
Commit
d86ce40
β€’
1 Parent(s): b3aa6b8

Update pages/2_😈_BlackBox_and_WhiteBox_Attacks.py

Browse files
pages/2_😈_BlackBox_and_WhiteBox_Attacks.py CHANGED
@@ -37,7 +37,7 @@ st.sidebar.markdown(
37
  )
38
 
39
  # title
40
- st.title('Blackbox Attack')
41
 
42
  #
43
  # They only use the last logit for text generation, so only using the last one would be fine.
@@ -55,7 +55,7 @@ def run(model, tokenizer, embedidng_layer=None, _bar_text=None, bar=None, text='
55
  _input[k] = _input[k].to(device)
56
 
57
  ori_output = model(**_input)
58
-
59
  ori_output = ori_output['logits']
60
 
61
  ori_embedding = embedidng_layer(_input['input_ids']).detach()
 
37
  )
38
 
39
  # title
40
+ st.title('Attacks')
41
 
42
  #
43
  # They only use the last logit for text generation, so only using the last one would be fine.
 
55
  _input[k] = _input[k].to(device)
56
 
57
  ori_output = model(**_input)
58
+
59
  ori_output = ori_output['logits']
60
 
61
  ori_embedding = embedidng_layer(_input['input_ids']).detach()