diff --git a/cyclegan/models/test_model.py b/cyclegan/models/test_model.py
index 7b5aaea..65145f5 100644
--- a/cyclegan/models/test_model.py
+++ b/cyclegan/models/test_model.py
@@ -78,6 +78,8 @@ class TestModel(BaseModel):
         black = np.zeros((1, 3, image.size(2), image.size(3)))
         black = torch.FloatTensor(black).cuda()
         input_adv, perturb = pgd_attack.perturb(image, black)      
+
+        black = None
         
         return input_adv, perturb
 
diff --git a/cyclegan/test.py b/cyclegan/test.py
index 916d5bb..33cd4fa 100644
--- a/cyclegan/test.py
+++ b/cyclegan/test.py
@@ -88,7 +88,8 @@ if __name__ == '__main__':
         model.fake = None
         model.real = None
         model.fake_noattack = None
-        
+        data = None
+
     webpage.save()  # save the HTML