diff --git a/test_model.py b/test_model.py
index 6a3d971..796c7bd 100644
--- a/test_model.py
+++ b/test_model.py
@@ -1,6 +1,4 @@
 import torch
-from torchsummary import summary
-import numpy as np
 
 from nets import Model