changed some files

app.py

@@ -131,7 +131,7 @@ def predict():
         # Extract train, test, and test_size with defaults
         train = data.get('train', False)
         test = data.get('test', False)
-        test_size = float(data.get('test_size', 0.1))
+        test_size = float(data.get('train_size', 0.1))
 
         device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')