Update app.py

app.py

@@ -269,10 +269,11 @@ class CNN1D(nn.Module):
             nn.Conv1d(32, 64, 3, padding=1), nn.ReLU(),
             nn.AdaptiveAvgPool1d(1),
             nn.Flatten(),
-            nn.Linear(64, len(np.unique(y)))
+            nn.Linear(64, num_classes)
         )
     def forward(self, x): return self.net(x)
-
+        
+num_classes = len(np.unique(np.concatenate([y_train_raw, y_test_raw])))
 model = CNN1D()
 criterion = nn.CrossEntropyLoss()
 optimizer = optim.Adam(model.parameters(), lr=0.001)