Update app.py

app.py

@@ -45,7 +45,10 @@ def compute(df_table):
         x_train = torch.tensor(np.delete(train_table, y_column, axis=1).astype(np.float32))
         x_eval = torch.tensor(np.delete(eval_table, y_column, axis=1).astype(np.float32))
 
-        y_train = train_table[:, y_column]
+        #y_train = train_table[:, y_column]
+        y_train = np.array(train_table[:, y_column].tolist())
+
+
     except ValueError:
         return "⚠️ **Please only add numbers (to the inputs) or leave fields empty.**", None, None