Update app.py

app.py

@@ -209,10 +209,10 @@ if ("0" == "mycustom"):
 if ("0" != "mycustom"):
 	x_train = train_data[[i for i in range(len(myfields))]].values
 
-st.write(start,end)
-st.dataframe(features)
-
-y_train = features.iloc[start:end][[1]]
+if ("0" == "mycustom"):
+	y_train = features.iloc[start:end][[1]]
+if ("0" != "mycustom"):
+	y_train = features.iloc[start:end][[0]]
 
 sequence_length = int(past / step)
 x_end = len(val_data) - past - future