fix model name

seizure_detection.py

@@ -3,9 +3,9 @@ import torch.nn as nn
 import torch.nn.functional as F
 
 
-class SeizureDetectionCNN(nn.Module):
+class SeizureDetector(nn.Module):
     def init(self, num_classes=2):
-        super(SeizureDetectionCNN, self).init()
+        super(SeizureDetector, self).init()
         self.conv1= nn.Conv2d(1, 32, kernel_size=3, stride=1, padding=1) # 32, 32, 32
 
         self.pool= nn.MaxPool2d(kernel_size=2, stride=2) # 32, 16, 16