From 1710ae154403e45a14e1c91fe8ade64d44c994c9 Mon Sep 17 00:00:00 2001
From: ZEMINCHEN <157980234+jimmychen9@users.noreply.github.com>
Date: Mon, 20 Jan 2025 01:33:58 -0500
Subject: [PATCH] Update adv_image.py

---
 deepsecure/adv_image.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/deepsecure/adv_image.py b/deepsecure/adv_image.py
index 31b8b1d..4d16e03 100644
--- a/deepsecure/adv_image.py
+++ b/deepsecure/adv_image.py
@@ -19,6 +19,10 @@ def weights_init(m):
 
 
 class Adv_Gen:
+    # Initialize the adversarial generator class with essential components.
+    # `device`: Specifies the computation device (e.g., CPU or GPU).
+    # `model_extractor`: Pre-trained feature extraction model used to extract important features.
+    # `generator`: Neural network model responsible for generating adversarial examples.
     def __init__(self,
                  device,
                  model_extractor,