Update test/adversarial_test.py

test/adversarial_test.py

@@ -1,32 +1,29 @@
-# test/adversarial_test.py
+# test/adversarial-attack.py
 
 from art.estimators.language import HuggingFacePipeline
 from art.attacks.text import TextAttackExtraction, TextFooler
-from transformers import pipeline
-import os
 
-# Load your model pipeline from app.py
-# Replace this with your actual function if you wrapped it differently
-from app import generator  
+# Import your pipeline from app.py
+from app import generator
 
-# Wrap the HF pipeline with ART
+# Wrap Hugging Face pipeline with ART estimator
 art_model = HuggingFacePipeline(pipeline=generator)
 
 # Example prompt to test
 prompt = "Tell me a positive thing about flowers in Africa."
 
-# === Attack 1: TextAttackExtraction ===
+# ===== Attack 1: TextAttackExtraction =====
 textattack = TextAttackExtraction(estimator=art_model)
 adv_examples_extraction = textattack.generate([prompt])
 
-print("TextAttackExtraction Adversarial Examples:")
+print("=== TextAttackExtraction Adversarial Examples ===")
 for ex in adv_examples_extraction:
     print(ex)
 
-# === Attack 2: TextFooler ===
+# ===== Attack 2: TextFooler =====
 textfooler = TextFooler(estimator=art_model)
 adv_examples_fooler = textfooler.generate([prompt])
 
-print("\nTextFooler Adversarial Examples:")
+print("\n=== TextFooler Adversarial Examples ===")
 for ex in adv_examples_fooler:
     print(ex)