anonymous8 commited on
Commit
69faf51
1 Parent(s): fa6512e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -18,8 +18,8 @@ from textattack.attack_recipes import (
18
  CLARE2020,
19
  )
20
  from textattack.attack_results import SuccessfulAttackResult
21
- from utils import SentAttacker, get_agnews_example, get_sst2_example, get_amazon_example, get_imdb_example, diff_texts, \
22
- get_yahoo_example
23
 
24
  sent_attackers = {}
25
  tad_classifiers = {}
@@ -77,8 +77,8 @@ def generate_adversarial_example(dataset, attacker, text=None, label=None):
77
  text, label = get_sst2_example()
78
  elif "amazon" in dataset.lower():
79
  text, label = get_amazon_example()
80
- elif "yahoo" in dataset.lower():
81
- text, label = get_yahoo_example()
82
  elif "imdb" in dataset.lower():
83
  text, label = get_imdb_example()
84
 
 
18
  CLARE2020,
19
  )
20
  from textattack.attack_results import SuccessfulAttackResult
21
+ from utils import SentAttacker, get_agnews_example, get_sst2_example, get_amazon_example, get_imdb_example, diff_texts
22
+ # from utils import get_yahoo_example
23
 
24
  sent_attackers = {}
25
  tad_classifiers = {}
 
77
  text, label = get_sst2_example()
78
  elif "amazon" in dataset.lower():
79
  text, label = get_amazon_example()
80
+ # elif "yahoo" in dataset.lower():
81
+ # text, label = get_yahoo_example()
82
  elif "imdb" in dataset.lower():
83
  text, label = get_imdb_example()
84