Update app.py

app.py

@@ -185,7 +185,8 @@ train_data = train_data.reindex(columns=mycolumns)
 
 x_train = train_data[[i for i in range(myrangeend)]].values
 #y_train = features.iloc[start:end][[1]]
-y_train = features.reindex(columns=mycolumns).iloc[start:end][[1]]
+#y_train = features.reindex(columns=mycolumns).iloc[start:end][[1]]
+y_train = mycolumns
 
 sequence_length = int(past / step)
 x_end = len(val_data) - past - future
@@ -249,4 +250,3 @@ fig = plot()
 st.pyplot(fig)
 
 # %%
-