PFEemp2024 commited on
Commit
7a9762c
1 Parent(s): bf372ad

fixing the dataset bug

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -69,7 +69,7 @@ cache = set()
69
 
70
 
71
  def generate_adversarial_example(dataset, attacker, text=None, label=None):
72
- """if not text or text in cache:
73
  if "agnews" in dataset.lower():
74
  text, label = get_agnews_example()
75
  elif "sst2" in dataset.lower():
@@ -79,7 +79,7 @@ def generate_adversarial_example(dataset, attacker, text=None, label=None):
79
  # elif "yahoo" in dataset.lower():
80
  # text, label = get_yahoo_example()
81
  elif "imdb" in dataset.lower():
82
- text, label = get_imdb_example()"""
83
 
84
  cache.add(text)
85
 
 
69
 
70
 
71
  def generate_adversarial_example(dataset, attacker, text=None, label=None):
72
+ if not text or text in cache:
73
  if "agnews" in dataset.lower():
74
  text, label = get_agnews_example()
75
  elif "sst2" in dataset.lower():
 
79
  # elif "yahoo" in dataset.lower():
80
  # text, label = get_yahoo_example()
81
  elif "imdb" in dataset.lower():
82
+ text, label = get_imdb_example()
83
 
84
  cache.add(text)
85