Update handler.py

handler.py

@@ -8,20 +8,22 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 print('Device:', device)
 
 
+# Define model for the model class
+
+torch.manual_seed(42)
+
+model = nn.Sequential(
+    nn.Linear(12, 12),
+    nn.ReLU(),
+    nn.Linear(12, 6),
+    nn.ReLU(),
+    nn.Linear(6, 1),
+    nn.Sigmoid()
+)
+
 # Define model class
 class MyModel(nn.Module, PyTorchModelHubMixin):
 
-    torch.manual_seed(42)
-
-    model = nn.Sequential(
-        nn.Linear(12, 12),
-        nn.ReLU(),
-        nn.Linear(12, 6),
-        nn.ReLU(),
-        nn.Linear(6, 1),
-        nn.Sigmoid()
-    )
-    
     def __init__(self):
         super().__init__()  # Initialize nn.Module
         self.model = model