diff --git a/model.py b/model.py
index 0a37c08..1c24dc2 100644
--- a/model.py
+++ b/model.py
@@ -1,4 +1,5 @@
 import torch
+from torch import nn
 from torchaudio.models import DeepSpeech
 import pytorch_lightning as pl
 from utils import TextProcess, CTCDecoder