Update preprocess_test.py

preprocess_test.py

@@ -10,6 +10,25 @@ from huggingface_hub import hf_hub_download
 
 
 
+
+class Model(nn.Module):
+    def __init__(self, input_shape, num_classes):
+        super(Model, self).__init__()
+        self.fc1 = nn.Linear(input_shape, 1024)
+        self.bn1 = nn.BatchNorm1d(1024)
+        self.fc2 = nn.Linear(1024, 512)
+        self.bn2 = nn.BatchNorm1d(512)
+        self.fc3 = nn.Linear(512, 256)
+        self.bn3 = nn.BatchNorm1d(256)
+        self.fc4 = nn.Linear(256, num_classes)
+
+    def forward(self, x):
+        x = F.relu(self.bn1(self.fc1(x)))
+        x = F.relu(self.bn2(self.fc2(x)))
+        x = F.relu(self.bn3(self.fc3(x)))
+        x = self.fc4(x)
+        return x
+
 class Preprocess_Test:
     def __init__(self,df):