diff --git a/model/u2net.py b/model/u2net.py
index ece59e08..1ca0c038 100644
--- a/model/u2net.py
+++ b/model/u2net.py
@@ -1,6 +1,5 @@
 import torch
 import torch.nn as nn
-from torchvision import models
 import torch.nn.functional as F
 
 class REBNCONV(nn.Module):