diff --git a/environment.yml b/environment.yml
new file mode 100644
index 0000000..6507ad1
Binary files /dev/null and b/environment.yml differ
diff --git a/model.py b/model.py
index e69de29..5ac275b 100644
--- a/model.py
+++ b/model.py
@@ -0,0 +1,36 @@
+import torch
+import torch.nn as nn
+
+
+class _ConvWithPReLU(nn.Module):
+    def __init__(self, in_channels, out_channels, kernel_size, stride, padding=0):
+        super(_ConvWithPReLU, self).__init__()
+        self.conv = nn.Conv2d(in_channels, out_channels, kernel_size, stride, padding)
+        self.prelu = nn.PReLU()
+
+    def forward(self, x):
+        x = self.conv(x)
+        x = self.prelu(x)
+        return x
+
+
+class _TransConvWithPReLU(nn.Module):
+    def __init__(self, in_channels, out_channels, kernel_size, stride, activate=nn.PReLU(), padding=0):
+        super(_TransConvWithPReLU, self).__init__()
+        self.transconv = nn.ConvTranspose2d(in_channels, out_channels, kernel_size, stride, padding)
+        self.activate = activate
+
+    def forward(self, x):
+        x = self.transconv(x)
+        x = self.activate(x)
+        return x
+
+
+class Nomalization(nn.Module):
+    def __init__(self, in_channels):
+        super(Nomalization, self).__init__()
+        self.norm = nn.BatchNorm2d(in_channels)
+
+    def forward(self, x):
+        x = self.norm(x)
+        return x
diff --git a/requirements.txt b/requirements.txt
deleted file mode 100644
index e69de29..0000000