diff --git a/src/main.py b/src/main.py
index 52c25c2..5089d65 100644
--- a/src/main.py
+++ b/src/main.py
@@ -14,12 +14,6 @@
 ])
 
 trainset = datasets.MNIST('.', download=True, train=True, transform=transform)
-trainloader = DataLoader(trainset, batch_size=64, shuffle=True)
-
-# Step 2: Define the PyTorch Model
-class Net(nn.Module):
-trainloader = DataLoader(trainset, batch_size=64, shuffle=True)
-
 # Step 2: Define the PyTorch Model
 class Net(nn.Module):
     def __init__(self):