Upload 4 files

app.py

@@ -31,7 +31,7 @@ class AphasiaPredictionApp:
         self.latent_dim = MODEL_CONFIG['latent_dim']
         self.last_treatment_file = None  # Track the last treatment file used
         
-    def train_models(self, data_dir, latent_dim, nepochs, bsize):
+    def train_models(self, data_dir, latent_dim, nepochs, bsize, num_samples):
         """
         Train VAE and Random Forest models
         """
@@ -2238,7 +2238,8 @@ def create_interface():
                         data_dir=data_dir,
                         latent_dim=latent_dim,
                         nepochs=nepochs,
-                        bsize=bsize
+                        bsize=bsize,
+                        num_samples=num_samples
                     )
                     
                     # Store results in app_state for the next tabs
@@ -2263,7 +2264,8 @@ def create_interface():
                         data_dir=data_dir,
                         latent_dim=latent_dim,
                         nepochs=nepochs,
-                        bsize=bsize
+                        bsize=bsize,
+                        num_samples=num_samples
                     )
                     
                     # Store results in app_state